• Home
• com.github.wolf480pl
• mias4j
• 0.1.0
• source code
• SandboxAdapter.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.wolf480pl.mias4j.core.rewrite.SandboxAdapter Maven / Gradle / Ivy

/*
 * Copyright (c) 2014 Wolf480pl 
 * This program is licensed under the GNU Lesser General Public License.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */
package com.github.wolf480pl.mias4j.core.rewrite;

import static org.objectweb.asm.Type.getType;

import java.lang.invoke.CallSite;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.Handle;
import org.objectweb.asm.Label;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
import org.objectweb.asm.commons.AnalyzerAdapter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.github.wolf480pl.mias4j.core.InvocationType;
import com.github.wolf480pl.mias4j.core.runtime.ArgumentPack;
import com.github.wolf480pl.mias4j.core.runtime.Bootstraps;
import com.github.wolf480pl.mias4j.util.SequenceMethodVisitor;
import com.github.wolf480pl.mias4j.util.WrappedCheckedException;

public class SandboxAdapter extends ClassVisitor {
    public static final String INIT = "";
    protected static final Logger LOG = LoggerFactory.getLogger(SandboxAdapter.class);

    private final RewritePolicy policy;
    private Type clazz;

    public SandboxAdapter(ClassVisitor cv) {
        this(cv, BlindPolicy.ALWAYS_INTERCEPT);
    }

    public SandboxAdapter(ClassVisitor cv, RewritePolicy policy) {
        super(Opcodes.ASM5, cv);
        this.policy = policy;
    }

    @Override
    public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
        this.clazz = Type.getObjectType(name);
        super.visit(version, access, name, signature, superName, interfaces);
    }

    @Override
    public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
        MethodAdapter ma = new MethodAdapter(cv.visitMethod(access, name, desc, signature, exceptions), policy, clazz, name.equals(INIT));
        AnalyzerAdapter analyzer = new AnalyzerAdapter(clazz.getInternalName(), access, name, desc, ma);
        ma.setAnalyzer(analyzer);
        return analyzer;
    }

    public static class MethodAdapter extends SequenceMethodVisitor {
        public static final String WRAPINVOKE_NAME = Bootstraps.WRAPINVOKE_NAME;
        public static final String WRAPINVOKE_DESC = Type.getMethodDescriptor(getType(CallSite.class), getType(MethodHandles.Lookup.class), getType(String.class), getType(MethodType.class),
                Type.INT_TYPE, getType(String.class), getType(MethodType.class));

        public static final String WRAPCONSTRUCTOR_NAME = Bootstraps.WRAPCONSTRUCTOR_NAME;
        public static final String WRAPCONSTRUCTOR_DESC = Type.getMethodDescriptor(getType(CallSite.class), getType(MethodHandles.Lookup.class), getType(String.class), getType(MethodType.class),
                getType(String.class), getType(MethodType.class));

        public static final String WRAPSUPERCONSTRUCTORARGS_NAME = Bootstraps.WRAPSUPERCONSTRUCTORARGS_NAME;
        public static final String WRAPSUPERCONSTRUCTORARGS_DESC = Type.getMethodDescriptor(getType(CallSite.class), getType(MethodHandles.Lookup.class), getType(String.class),
                getType(MethodType.class), getType(String.class), getType(MethodType.class));

        public static final String WRAPSUPERCONSTRUCTORRES_NAME = Bootstraps.WRAPSUPERCONSTRUCTORRES_NAME;
        public static final String WRAPSUPERCONSTRUCTORRES_DESC = Type.getMethodDescriptor(getType(CallSite.class), getType(MethodHandles.Lookup.class), getType(String.class),
                getType(MethodType.class), getType(String.class), getType(MethodType.class));

        public static final String WRAPHANDLE_NAME = Bootstraps.WRAPHANDLE_NAME;
        public static final String WRAPHANDLE_DESC = Type.getMethodDescriptor(getType(CallSite.class), getType(MethodHandles.Lookup.class), getType(String.class), getType(MethodType.class),
                Type.INT_TYPE, getType(String.class), getType(MethodType.class));

        public static final String WRAPDYNAMIC_NAME = Bootstraps.WRAPDYNAMIC_NAME;
        public static final String WRAPDYNAMIC_DESC = Type.getMethodDescriptor(getType(CallSite.class), getType(MethodHandles.Lookup.class), getType(String.class), getType(MethodType.class),
                Type.INT_TYPE, getType(String.class), getType(String.class), getType(MethodType.class), getType(Object[].class));

        public static final String[] ARGPACK_NAMES = makeArgPackNameTable();
        public static final String[] ARGPACK_DESCS;

        private final RewritePolicy policy;
        private final Type clazz;
        private final boolean constructor;
        private AnalyzerAdapter analyzer;
        private Set removedNews = new HashSet<>();
        private List labels = new ArrayList<>(3);

        public MethodAdapter(MethodVisitor mv, RewritePolicy policy, Type clazz, boolean constructor) {
            super(mv);
            this.constructor = constructor;
            this.clazz = clazz;
            this.policy = policy;
        }

        public void setAnalyzer(AnalyzerAdapter analyzer) {
            this.analyzer = analyzer;
        }

        @Override
        public void visitMethodInsn(int opcode, String owner, String name, String desc, boolean itf) {
            labels.clear();

            Type ownerType = Type.getObjectType(owner);
            Type methType = Type.getMethodType(desc);

            int argAndReturnSizes = methType.getArgumentsAndReturnSizes();
            // FIXME: This will crash when rewriting unreachable code (analyzer.stack == null)
            int arg0idx = analyzer.stack.size() - (argAndReturnSizes >> 2);

            InvocationType invtype;
            if (opcode == Opcodes.INVOKESPECIAL && name.equals(INIT)) {
                if (constructor && analyzer.stack.get(arg0idx) == Opcodes.UNINITIALIZED_THIS) {
                    invtype = InvocationType.INVOKESUPERINITSPECIAL;
                } else {
                    invtype = InvocationType.INVOKENEWSPECIAL;
                }
            } else {
                invtype = InvocationType.fromInstruction(opcode);
            }

            boolean should = true;
            boolean decidedOnNew = (invtype == InvocationType.INVOKENEWSPECIAL) && removedNews.contains(analyzer.stack.get(arg0idx));

            if (!decidedOnNew) {
                try {
                    should = policy.shouldIntercept(clazz, invtype, ownerType, name, methType);
                } catch (RewriteAbortException e) {
                    throw new WrappedCheckedException(e);
                }
            }
            if (!should) {
                mv.visitMethodInsn(opcode, owner, name, desc, itf);
                return;
            }

            LOG.debug("methType: " + methType);
            LOG.debug("arg&ret size: " + argAndReturnSizes + " stack size: " + analyzer.stack.size() + " arg0idx: " + arg0idx);
            if (invtype == InvocationType.INVOKESUPERINITSPECIAL) {
                int thisLocal;
                int freeLocal = analyzer.locals.size();
                if (analyzer.locals.get(0) == Opcodes.UNINITIALIZED_THIS) {
                    thisLocal = 0;
                } else {
                    thisLocal = -1;
                }

                Type nt = Type.getMethodType(getType(ArgumentPack.class), methType.getArgumentTypes());

                // Filter the arguments
                mv.visitInvokeDynamicInsn("init", nt.getDescriptor(), new Handle(Opcodes.H_INVOKESTATIC, Type.getInternalName(Bootstraps.class), WRAPSUPERCONSTRUCTORARGS_NAME,
                        WRAPSUPERCONSTRUCTORARGS_DESC), ownerType.getClassName(), methType);
                int packLocal = freeLocal++;
                mv.visitVarInsn(Opcodes.ASTORE, packLocal);
                if (thisLocal < 0) {
                    thisLocal = freeLocal++;
                    mv.visitVarInsn(Opcodes.ASTORE, thisLocal);
                    mv.visitVarInsn(Opcodes.ALOAD, thisLocal);
                }

                for (Type argType : methType.getArgumentTypes()) {
                    if (argType.getSort() == Type.VOID) {
                        continue;
                    }
                    mv.visitVarInsn(Opcodes.ALOAD, packLocal);
                    mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, Type.getInternalName(ArgumentPack.class), ARGPACK_NAMES[argType.getSort()], ARGPACK_DESCS[argType.getSort()], false);
                    if (ARGPACK_DESCS[argType.getSort()] == ARGPACK_NEXTOBJ_DESC) {
                        mv.visitTypeInsn(Opcodes.CHECKCAST, argType.getInternalName());
                    }
                }

                // Call the original superinitializer
                mv.visitMethodInsn(opcode, owner, name, desc, itf);

                // Give the policy a chance to inspect the outcome of the superinitializer call, and possibly throw something
                mv.visitVarInsn(Opcodes.ALOAD, thisLocal);
                mv.visitInvokeDynamicInsn("init", Type.getMethodDescriptor(Type.VOID_TYPE, ownerType), new Handle(Opcodes.H_INVOKESTATIC, Type.getInternalName(Bootstraps.class),
                        WRAPSUPERCONSTRUCTORRES_NAME, WRAPSUPERCONSTRUCTORRES_DESC), ownerType.getClassName(), methType);
                return;
            }

            if (invtype == InvocationType.INVOKENEWSPECIAL) {
                Type nt = Type.getMethodType(ownerType, methType.getArgumentTypes());
                desc = nt.getDescriptor();

                mv.visitInvokeDynamicInsn("init", desc, new Handle(Opcodes.H_INVOKESTATIC, Type.getInternalName(Bootstraps.class), WRAPCONSTRUCTOR_NAME, WRAPCONSTRUCTOR_DESC),
                        ownerType.getClassName(), methType);

                // stack: ... uninitialized returned
                int local = analyzer.locals.size();
                mv.visitVarInsn(Opcodes.ASTORE, local);
                // stack: ... uninitialized
                mv.visitInsn(Opcodes.POP);
                // stack: ...

                Label uninitialized = (Label) analyzer.stack.get(arg0idx);
                int currentStack = arg0idx - 1;

                // remove all occurrences of the uninitialized reference from stack...
                while (currentStack >= 0 && analyzer.stack.get(currentStack) == uninitialized) {
                    mv.visitInsn(Opcodes.POP);
                    --currentStack;
                }
                int lowestStack = currentStack;

                int deepestRef = -1;
                for (int i = 0; i < lowestStack; ++i) {
                    if (analyzer.stack.get(i) == uninitialized) {
                        deepestRef = i;
                        break;
                    }
                }
                if (deepestRef >= 0) {
                    findReplaceInStack(uninitialized, local, local + 1, deepestRef, currentStack);
                }

                // ...and replace them with the return value of our method handle
                while (currentStack < arg0idx - 1) {
                    mv.visitVarInsn(Opcodes.ALOAD, local);
                    ++currentStack;
                }

                // replace all occurrences of the uninitialized reference in locals with the return value of our metod handle
                for (int i = 0; i < analyzer.locals.size(); ++i) {
                    if (analyzer.locals.get(i) == uninitialized) {
                        mv.visitVarInsn(Opcodes.ALOAD, local);
                        mv.visitVarInsn(Opcodes.ASTORE, i);
                    }
                }

                return;
            }

            if (opcode != Opcodes.INVOKESTATIC) {
                Type[] argTypes = methType.getArgumentTypes();
                Type[] newArgTypes = new Type[argTypes.length + 1];
                newArgTypes[0] = ownerType;
                System.arraycopy(argTypes, 0, newArgTypes, 1, argTypes.length);
                Type nt = Type.getMethodType(methType.getReturnType(), newArgTypes);
                desc = nt.getDescriptor();
            }

            mv.visitInvokeDynamicInsn(name, desc,
                    new Handle(Opcodes.H_INVOKESTATIC, Type.getInternalName(Bootstraps.class), WRAPINVOKE_NAME, WRAPINVOKE_DESC),
                    invtype.id(), ownerType.getClassName(), methType);
        }

        @Override
        public void visitInvokeDynamicInsn(String name, String desc, Handle bsm, Object... bsmArgs) {
            labels.clear();

            Type methType = Type.getMethodType(desc);
            Handle newBsm;
            List