org.evosuite.instrumentation.ComparisonTransformation.java Source code

Introduction

Here is the source code for org.evosuite.instrumentation.ComparisonTransformation.java

Source

/**
 * Copyright (C) 2011,2012 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite is free software: you can redistribute it and/or modify it under the
 * terms of the GNU Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 *
 * EvoSuite 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 Public License for more details.
 *
 * You should have received a copy of the GNU Public License along with
 * EvoSuite. If not, see <http://www.gnu.org/licenses/>.
 */
/**
 * 
 */
package org.evosuite.instrumentation;

import java.util.List;

import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.AbstractInsnNode;
import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.InsnList;
import org.objectweb.asm.tree.InsnNode;
import org.objectweb.asm.tree.MethodInsnNode;
import org.objectweb.asm.tree.MethodNode;

/**
 * <p>ComparisonTransformation class.</p>
 *
 * @author Gordon Fraser
 */
public class ComparisonTransformation {

    ClassNode cn;

    /**
     * <p>Constructor for ComparisonTransformation.</p>
     *
     * @param cn a {@link org.objectweb.asm.tree.ClassNode} object.
     */
    public ComparisonTransformation(ClassNode cn) {
        this.cn = cn;
    }

    /**
     * <p>transform</p>
     *
     * @return a {@link org.objectweb.asm.tree.ClassNode} object.
     */
    @SuppressWarnings("unchecked")
    public ClassNode transform() {
        List<MethodNode> methodNodes = cn.methods;
        for (MethodNode mn : methodNodes) {
            transformMethod(mn);
        }
        return cn;
    }

    /**
     * <p>transformMethod</p>
     *
     * @param mn a {@link org.objectweb.asm.tree.MethodNode} object.
     */
    public void transformMethod(MethodNode mn) {
        AbstractInsnNode node = mn.instructions.getFirst();
        while (node != mn.instructions.getLast()) {
            AbstractInsnNode next = node.getNext();
            if (node instanceof InsnNode) {
                InsnNode in = (InsnNode) node;
                if (in.getOpcode() == Opcodes.LCMP) {
                    insertLongComparison(in, mn.instructions);
                    TransformationStatistics.transformedComparison();
                } else if (in.getOpcode() == Opcodes.DCMPG) {
                    // TODO: Check treatment of NaN
                    TransformationStatistics.transformedComparison();
                    insertDoubleComparison(in, mn.instructions);
                } else if (in.getOpcode() == Opcodes.DCMPL) {
                    TransformationStatistics.transformedComparison();
                    insertDoubleComparison(in, mn.instructions);
                } else if (in.getOpcode() == Opcodes.FCMPG) {
                    TransformationStatistics.transformedComparison();
                    insertFloatComparison(in, mn.instructions);
                } else if (in.getOpcode() == Opcodes.FCMPL) {
                    TransformationStatistics.transformedComparison();
                    insertFloatComparison(in, mn.instructions);
                }
            }
            node = next;
        }
    }

    private void insertLongComparison(AbstractInsnNode position, InsnList list) {
        MethodInsnNode get = new MethodInsnNode(Opcodes.INVOKESTATIC, Type.getInternalName(BooleanHelper.class),
                "longSub", Type.getMethodDescriptor(Type.INT_TYPE, new Type[] { Type.LONG_TYPE, Type.LONG_TYPE }));
        list.insert(position, get);
        list.remove(position);
    }

    private void insertFloatComparison(AbstractInsnNode position, InsnList list) {
        MethodInsnNode get = new MethodInsnNode(Opcodes.INVOKESTATIC, Type.getInternalName(BooleanHelper.class),
                "floatSub",
                Type.getMethodDescriptor(Type.INT_TYPE, new Type[] { Type.FLOAT_TYPE, Type.FLOAT_TYPE }));
        list.insert(position, get);
        list.remove(position);
    }

    private void insertDoubleComparison(AbstractInsnNode position, InsnList list) {
        MethodInsnNode get = new MethodInsnNode(Opcodes.INVOKESTATIC, Type.getInternalName(BooleanHelper.class),
                "doubleSub",
                Type.getMethodDescriptor(Type.INT_TYPE, new Type[] { Type.DOUBLE_TYPE, Type.DOUBLE_TYPE }));
        list.insert(position, get);
        list.remove(position);
    }
}