Example usage for org.objectweb.asm.tree ClassNode accept

Introduction

In this page you can find the example usage for org.objectweb.asm.tree ClassNode accept.

Prototype

public void accept(final ClassVisitor classVisitor) 

Source Link

Document

Makes the given class visitor visit this class.

Usage

From source file:net.sandius.rembulan.compiler.gen.asm.ASMBytecodeEmitter.java

License:Apache License

private byte[] classNodeToBytes(ClassNode classNode) {
    ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_MAXS);
    classNode.accept(writer);
    byte[] bytes = writer.toByteArray();

    // verify bytecode

    if (verifyAndPrint) {
        ClassReader reader = new ClassReader(bytes);
        ClassVisitor tracer = new TraceClassVisitor(new PrintWriter(System.out));
        ClassVisitor checker = new CheckClassAdapter(tracer, true);
        reader.accept(checker, 0);//from  w w  w  .  j a  v a 2 s. c om
    }

    return bytes;
}

---

From source file:nova.core.wrapper.mc.forge.v17.asm.lib.ASMHelper.java

License:Open Source License

public static byte[] createBytes(ClassNode cnode, int flags) {
    ClassWriter cw = new CC_ClassWriter(flags);
    cnode.accept(cw);
    return cw.toByteArray();
}

---

From source file:org.apache.drill.exec.compile.AsmUtil.java

License:Apache License

/**
 * Check to see if a class is well-formed.
 *
 * @param logger the logger to write to if a problem is found
 * @param logTag a tag to print to the log if a problem is found
 * @param classNode the class to check//from   w w  w .j  a  va2  s .  c  o  m
 * @return true if the class is ok, false otherwise
 */
public static boolean isClassOk(final Logger logger, final String logTag, final ClassNode classNode) {
    final StringWriter sw = new StringWriter();
    final ClassWriter verifyWriter = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
    classNode.accept(verifyWriter);
    final ClassReader ver = new ClassReader(verifyWriter.toByteArray());
    try {
        DrillCheckClassAdapter.verify(ver, false, new PrintWriter(sw));
    } catch (final Exception e) {
        logger.info("Caught exception verifying class:");
        logClass(logger, logTag, classNode);
        throw e;
    }
    final String output = sw.toString();
    if (!output.isEmpty()) {
        logger.info("Invalid class:\n" + output);
        return false;
    }

    return true;
}

---

From source file:org.apache.drill.exec.compile.AsmUtil.java

License:Apache License

/**
 * Write a class to the log.//from w  w  w .  j  a va  2  s .  c  o m
 *
 * <p>
 * Writes at level TRACE.
 *
 * @param logger
 *          the logger to write to
 * @param logTag
 *          a tag to print to the log
 * @param classNode
 *          the class
 */
public static void logClass(final Logger logger, final String logTag, final ClassNode classNode) {
    if (logger.isTraceEnabled()) {
        logger.trace(logTag);
        final StringWriter stringWriter = new StringWriter();
        final PrintWriter printWriter = new PrintWriter(stringWriter);
        final TraceClassVisitor traceClassVisitor = new TraceClassVisitor(printWriter);
        classNode.accept(traceClassVisitor);
        logger.trace(stringWriter.toString());
    }
}

---

From source file:org.apache.drill.exec.compile.MergeAdapter.java

License:Apache License

private static final void check(ClassNode node) {
    Exception e = null;/*from w w  w  .  j a va  2 s. c o m*/
    String error = "";

    try {
        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
        ClassVisitor cv = new CheckClassAdapter(cw, true);
        node.accept(cv);

        StringWriter sw = new StringWriter();
        PrintWriter pw = new PrintWriter(sw);
        CheckClassAdapter.verify(new ClassReader(cw.toByteArray()), false, pw);

        error = sw.toString();
    } catch (Exception ex) {
        e = ex;
    }

    if (!error.isEmpty() || e != null) {
        StringWriter sw2 = new StringWriter();
        PrintWriter pw2 = new PrintWriter(sw2);
        TraceClassVisitor v = new TraceClassVisitor(pw2);
        node.accept(v);
        if (e != null) {
            throw new RuntimeException(
                    "Failure validating class.  ByteCode: \n" + sw2.toString() + "\n\n====ERRROR====\n" + error,
                    e);
        } else {
            throw new RuntimeException("Failure validating class.  ByteCode: \n" + sw2.toString()
                    + "\n\n====ERRROR====\n" + error);
        }
    }
}

---

From source file:org.apache.drill.exec.compile.TestClassTransformation.java

License:Apache License

@Test // DRILL-6524
public void testScalarReplacementInCondition() throws Exception {
    ClassTransformer.ClassNames classNames = new ClassTransformer.ClassNames(
            "org.apache.drill.CompileClassWithIfs");
    String entireClass = DrillFileUtils
            .getResourceAsString(DrillFileUtils.SEPARATOR + classNames.slash + ".java");

    sessionOptions.setLocalOption(ClassCompilerSelector.JAVA_COMPILER_DEBUG_OPTION, false);

    List<String> compilers = Arrays.asList(ClassCompilerSelector.CompilerPolicy.JANINO.name(),
            ClassCompilerSelector.CompilerPolicy.JDK.name());
    for (String compilerName : compilers) {
        sessionOptions.setLocalOption(ClassCompilerSelector.JAVA_COMPILER_OPTION, compilerName);

        QueryClassLoader queryClassLoader = new QueryClassLoader(config, sessionOptions);

        byte[][] implementationClasses = queryClassLoader.getClassByteCode(classNames, entireClass);

        ClassNode originalClass = AsmUtil.classFromBytes(implementationClasses[0], ClassReader.EXPAND_FRAMES);

        ClassNode transformedClass = new ClassNode();
        DrillCheckClassAdapter mergeGenerator = new DrillCheckClassAdapter(CompilationConfig.ASM_API_VERSION,
                new CheckClassVisitorFsm(CompilationConfig.ASM_API_VERSION, transformedClass), true);
        originalClass.accept(new ValueHolderReplacementVisitor(mergeGenerator, true));

        if (!AsmUtil.isClassOk(logger, classNames.dot, transformedClass)) {
            throw new IllegalStateException(
                    String.format("Problem found after transforming %s", classNames.dot));
        }/*from   w  w w  .  j a v a 2  s  . c o  m*/
        ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
        transformedClass.accept(writer);
        byte[] outputClass = writer.toByteArray();

        queryClassLoader.injectByteCode(classNames.dot, outputClass);
        Class<?> transformedClazz = queryClassLoader.findClass(classNames.dot);
        transformedClazz.getMethod("doSomething").invoke(null);
    }
}

---

From source file:org.apache.felix.scrplugin.helper.ClassModifier.java

License:Apache License

/**
 * Add bind/unbind methods/*from  w ww .j a  va  2  s  .  c om*/
 * @param className       The class name in which the methods are injected
 * @param referenceName   Name of the reference
 * @param fieldName       Name of the field
 * @param typeName        Name of the type
 * @param createBind      Name of the bind method or null
 * @param createUnbind    Name of the unbind method or null
 * @param outputDirectory Output directory where the class file is stored
 * @throws SCRDescriptorException
 */
public static void addMethods(final String className, final String referenceName, final String fieldName,
        final String typeName, final boolean createBind, final boolean createUnbind,
        final ClassLoader classLoader, final String outputDirectory, final Log logger)
        throws SCRDescriptorException {
    // now do byte code manipulation
    final String fileName = outputDirectory + File.separatorChar + className.replace('.', File.separatorChar)
            + ".class";
    final ClassNode cn = new ClassNode();
    try {
        final ClassReader reader = new ClassReader(new FileInputStream(fileName));
        reader.accept(cn, 0);

        // For target Java7 and above use: ClassWriter.COMPUTE_MAXS  | ClassWriter.COMPUTE_FRAMES
        final int mask = (cn.version > 50 ? ClassWriter.COMPUTE_MAXS | ClassWriter.COMPUTE_FRAMES : 0);
        final ClassWriter writer = new ClassWriter(mask) {

            @Override
            protected String getCommonSuperClass(final String type1, final String type2) {
                Class<?> c, d;
                try {
                    c = classLoader.loadClass(type1.replace('/', '.'));
                    d = classLoader.loadClass(type2.replace('/', '.'));
                } catch (final Exception e) {
                    throw new RuntimeException(e.toString(), e);
                }
                if (c.isAssignableFrom(d)) {
                    return type1;
                }
                if (d.isAssignableFrom(c)) {
                    return type2;
                }
                if (c.isInterface() || d.isInterface()) {
                    return "java/lang/Object";
                }
                do {
                    c = c.getSuperclass();
                } while (!c.isAssignableFrom(d));
                return c.getName().replace('.', '/');
            }

        };

        cn.accept(writer);
        if (createBind) {
            logger.debug("Adding bind " + className + " " + fieldName);

            createMethod(writer, className, referenceName, fieldName, typeName, true);
        }
        if (createUnbind) {
            logger.debug("Adding unbind " + className + " " + fieldName);

            createMethod(writer, className, referenceName, fieldName, typeName, false);
        }

        final FileOutputStream fos = new FileOutputStream(fileName);
        fos.write(writer.toByteArray());
        fos.close();
    } catch (final Exception e) {
        throw new SCRDescriptorException("Unable to add methods to " + className, typeName, e);
    }
}

---

From source file:org.apache.felix.scrplugin.tags.qdox.QDoxJavaClassDescription.java

License:Apache License

/**
 * @see org.apache.felix.scrplugin.tags.ModifiableJavaClassDescription#addMethods(java.lang.String, java.lang.String, boolean, boolean)
 *//* w w w .jav a  2  s . c  o m*/
public void addMethods(final String propertyName, final String className, final boolean createBind,
        final boolean createUnbind) throws SCRDescriptorException {
    // now do byte code manipulation
    final String targetDirectory = this.manager.getOutputDirectory();
    final String fileName = targetDirectory + File.separatorChar
            + this.getName().replace('.', File.separatorChar) + ".class";
    final ClassNode cn = new ClassNode();
    try {
        final ClassReader reader = new ClassReader(new FileInputStream(fileName));
        reader.accept(cn, 0);

        final ClassWriter writer = new ClassWriter(0);

        // remove existing implementation von previous builds
        final ClassAdapter adapter = new ClassAdapter(writer) {

            protected final String bindMethodName = "bind" + propertyName.substring(0, 1).toUpperCase()
                    + propertyName.substring(1);
            protected final String unbindMethodName = "unbind" + propertyName.substring(0, 1).toUpperCase()
                    + propertyName.substring(1);
            protected final String description = "(L" + className.replace('.', '/') + ";)V";

            /**
             * @see org.objectweb.asm.ClassAdapter#visitMethod(int, java.lang.String, java.lang.String, java.lang.String, java.lang.String[])
             */
            public MethodVisitor visitMethod(int access, String name, String desc, String signature,
                    String[] exceptions) {
                if (createBind && name.equals(bindMethodName) && description.equals(desc)) {
                    return null;
                }
                if (createUnbind && name.equals(unbindMethodName) && description.equals(desc)) {
                    return null;
                }
                return super.visitMethod(access, name, desc, signature, exceptions);
            }

        };

        cn.accept(adapter);
        if (createBind) {
            this.createMethod(writer, propertyName, className, true);
        }
        if (createUnbind) {
            this.createMethod(writer, propertyName, className, false);
        }

        final FileOutputStream fos = new FileOutputStream(fileName);
        fos.write(writer.toByteArray());
        fos.close();
    } catch (Exception e) {
        throw new SCRDescriptorException("Unable to add methods to " + this.getName(), className, 0, e);
    }
}

---

From source file:org.blockartistry.DynSurround.asm.Transformer.java

License:MIT License

private byte[] transformEntityRenderer(final byte[] classBytes) {
    final String names[] = { "func_78484_h", "addRainParticles" };
    final String targetName = "addRainParticles";

    final ClassReader cr = new ClassReader(classBytes);
    final ClassNode cn = new ClassNode(ASM5);
    cr.accept(cn, 0);//from   w ww  .  java 2  s.c o  m

    for (final MethodNode m : cn.methods) {
        if (isOneOf(m.name, names)) {
            logger.debug("Hooking " + m.name);
            final InsnList list = new InsnList();
            list.add(new VarInsnNode(ALOAD, 0));
            final String sig = "(Lnet/minecraft/client/renderer/EntityRenderer;)V";
            list.add(new MethodInsnNode(INVOKESTATIC,
                    "org/blockartistry/DynSurround/client/weather/RenderWeather", targetName, sig, false));
            list.add(new InsnNode(RETURN));
            m.instructions.insertBefore(m.instructions.getFirst(), list);
        }
    }

    final ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS);
    cn.accept(cw);
    return cw.toByteArray();
}

---

From source file:org.blockartistry.DynSurround.asm.Transformer.java

License:MIT License

private byte[] transformWorldServer(final byte[] classBytes) {
    final String names[] = { "func_73051_P", "resetRainAndThunder" };
    final String targetName = "resetRainAndThunder";

    final ClassReader cr = new ClassReader(classBytes);
    final ClassNode cn = new ClassNode(ASM5);
    cr.accept(cn, 0);//from   w w w. j  a  va2s.c  om

    for (final MethodNode m : cn.methods) {
        if (isOneOf(m.name, names)) {
            logger.debug("Hooking " + m.name);
            InsnList list = new InsnList();
            list.add(new VarInsnNode(ALOAD, 0));
            final String sig = "(Lnet/minecraft/world/WorldServer;)V";
            list.add(new MethodInsnNode(INVOKESTATIC, "org/blockartistry/DynSurround/server/PlayerSleepHandler",
                    targetName, sig, false));
            list.add(new InsnNode(RETURN));
            m.instructions.insertBefore(m.instructions.getFirst(), list);
            break;
        }
    }

    final ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS);
    cn.accept(cw);
    return cw.toByteArray();
}

---

• «
• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16
• 17
• 18
• 19
• 20
• 21
• 22
• 23
• 24
• »