CrashReportTransformer.java 2.4 KB
/*
 * This file is part of LiteLoader.
 * Copyright (C) 2012-16 Adam Mummery-Smith
 * All Rights Reserved.
 */
package com.mumfrey.liteloader.client.transformers;

import java.util.ListIterator;

import org.spongepowered.asm.lib.Opcodes;
import org.spongepowered.asm.lib.tree.AbstractInsnNode;
import org.spongepowered.asm.lib.tree.ClassNode;
import org.spongepowered.asm.lib.tree.InsnList;
import org.spongepowered.asm.lib.tree.MethodInsnNode;
import org.spongepowered.asm.lib.tree.MethodNode;
import org.spongepowered.asm.lib.tree.VarInsnNode;

import com.mumfrey.liteloader.core.runtime.Obf;
import com.mumfrey.liteloader.transformers.ClassTransformer;

public class CrashReportTransformer extends ClassTransformer
{
    @Override
    public byte[] transform(String name, String transformedName, byte[] basicClass)
    {
        if (basicClass != null && (Obf.CrashReport$6.name.equals(name) || Obf.CrashReport$6.obf.equals(name)))
        {
            try
            {
                return this.transformCallableJVMFlags(basicClass);
            }
            catch (Exception ex) {}
        }

        return basicClass;
    }

    /**
     * Inject the additional callback for populating the crash report into the
     * CallableJVMFlags class.
     * 
     * @param basicClass basic class
     * @return transformed class
     */
    private byte[] transformCallableJVMFlags(byte[] basicClass)
    {
        ClassNode classNode = this.readClass(basicClass, true);

        for (MethodNode method : classNode.methods)
        {
            if ("<init>".equals(method.name))
            {
                this.transformCallableJVMFlagsConstructor(method);
            }
        }

        return this.writeClass(classNode);
    }

    /**
     * @param ctor
     */
    public void transformCallableJVMFlagsConstructor(MethodNode ctor)
    {
        InsnList code = new InsnList();
        code.add(new VarInsnNode(Opcodes.ALOAD, 1));
        code.add(new MethodInsnNode(Opcodes.INVOKESTATIC, "com/mumfrey/liteloader/core/LiteLoader", "populateCrashReport",
                "(Ljava/lang/Object;)V", false));

        ListIterator<AbstractInsnNode> insns = ctor.instructions.iterator();
        while (insns.hasNext())
        {
            AbstractInsnNode insnNode = insns.next();
            if (insnNode.getOpcode() == Opcodes.RETURN)
            {
                ctor.instructions.insertBefore(insnNode, code);
            }
        }
    }
}