IndexParamAccessCheckAdapter.java

/*
* JBoss, Home of Professional Open Source
* Copyright 2010 Red Hat and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This 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 2.1 of
* the License, or (at your option) any later version.
*
* This software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*
* @authors Andrew Dinn
*/
package org.jboss.byteman.agent.adapter;

import org.jboss.byteman.agent.Location;
import org.jboss.byteman.agent.TransformContext;
import org.jboss.byteman.rule.type.Type;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;

/**
 * asm Adapter class used to check that the target method for a rule exists in a class
 */
public class IndexParamAccessCheckAdapter extends RuleCheckAdapter
{
     public IndexParamAccessCheckAdapter(ClassVisitor cv, TransformContext transformContext,
                                       int paramIdx, int flags, int count)
    {
        super(cv, transformContext);
        this.paramIdx = paramIdx;
        this.flags = flags;
        this.count = count;
    }

    public MethodVisitor visitMethod(
        final int access,
        final String name,
        final String desc,
        final String signature,
        final String[] exceptions)
    {
        MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
        if (matchTargetMethod(access, name, desc)) {
            int paramSlotIdx = Type.paramSlotIdx(access, desc, paramIdx);
            if (paramSlotIdx >= 0) {
                setVisited();
                return new IndexParamAccessCheckMethodAdapter(mv, getTransformContext(), paramSlotIdx, access, name, desc, signature, exceptions);
            }
        }

        return mv;
    }

    /**
     * a method visitor used to add a rule event trigger call to a method
     */

    // TODO fix this to deal with names which are actually integers i.e. $1, $2 etc
    // TODO probably best to use a different adapter
    private class IndexParamAccessCheckMethodAdapter extends RuleCheckMethodAdapter
    {
        private int access;
        private String name;
        private String descriptor;
        private String signature;
        private String[] exceptions;
        private int index;
        private int visitedCount;

        IndexParamAccessCheckMethodAdapter(MethodVisitor mv, TransformContext transformContext, int paramSlotIdx, int access, String name, String descriptor, String signature, String[] exceptions)
        {
            super(mv, transformContext, access, name, descriptor);
            this.access = access;
            this.name = name;
            this.descriptor = descriptor;
            this.signature = signature;
            this.exceptions = exceptions;
            // index -1 indicates that the var is not in scope
            this.index =  paramSlotIdx;
            visitedCount = 0;
        }

        private boolean matchCall(int opcode)
        {
            if (opcode < Opcodes.ISTORE)  {
                // it's a load
                return ((flags & Location.ACCESS_READ) != 0);
            } else {
                //it's a store
                return ((flags & Location.ACCESS_WRITE) != 0);
            }
        }

        @Override
        public void visitVarInsn(int opcode, int var) {
            if (var == index) {
                if ((count == 0 || (visitedCount < count)) && matchCall(opcode)) {
                    // a relevant invocation occurs in the called method
                    visitedCount++;
                    if (count == 0 || visitedCount == count) {
                        setTriggerPoint();
                    }
                }
            }
            super.visitVarInsn(opcode, var);
        }
        
        public void visitIincInsn(int var, int increment)
        {
            // IINC counts as a read and a write
            if (var == index) {
                if ((count == 0 || (visitedCount < count))) {
                    // a relevant invocation occurs in the called method
                    visitedCount++;
                    if (count == 0 || visitedCount == count) {
                        setTriggerPoint();
                    }
                }
            }
            super.visitIincInsn(var, increment);
        }
    }

    private int paramIdx;
    private int flags;
    private int count;
}