org.jvnet.animal_sniffer.SignatureBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of animal-sniffer Show documentation
The newest version!
package org.jvnet.animal_sniffer;

import org.objectweb.asm.ClassReader;
import org.objectweb.asm.FieldVisitor;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.commons.EmptyVisitor;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.util.HashSet;
import java.util.zip.GZIPOutputStream;

/**
 * Builds up a signature list from the given classes.
 *
 * @author Kohsuke Kawaguchi
 */
public class SignatureBuilder extends ClassFileVisitor {
    private boolean foundSome;
    public static void main(String[] args) throws IOException {
        SignatureBuilder builder = new SignatureBuilder(new FileOutputStream("signature"));
        builder.process(new File(System.getProperty("java.home"),"lib/rt.jar"));
        builder.close();
    }

    private final ObjectOutputStream oos;

    public SignatureBuilder(OutputStream out) throws IOException {
        oos = new ObjectOutputStream(new GZIPOutputStream(out));
    }

    public void close() throws IOException {
        oos.writeObject(null);   // EOF marker
        oos.close();
        if(!foundSome)  throw new IOException("No index is written");
    }

    protected void process(String name, InputStream image) throws IOException {
        System.out.println(name);
        foundSome=true;
        ClassReader cr = new ClassReader(image);
        SignatureVisitor v = new SignatureVisitor();
        cr.accept(v,0);
        v.end();
    }

    private class SignatureVisitor extends EmptyVisitor {
        Clazz clazz;

        public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
            this.clazz = new Clazz(name,new HashSet(),superName, interfaces);
        }

        public void end() throws IOException {
            oos.writeObject(clazz);
        }

        public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
            clazz.signatures.add(name+desc);
            return null;
        }

        public FieldVisitor visitField(int access, String name, String desc, String signature, Object value) {
            clazz.signatures.add(name+"#"+desc);
            return null;
        }
    }
}