org.evosuite.instrumentation.error.ErrorBranchInstrumenter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of evosuite-client Show documentation
The newest version!
/**
 * Copyright (C) 2010-2018 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 Lesser General Public License as published
 * by the Free Software Foundation, either version 3.0 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
 * Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.instrumentation.error;

import java.util.HashMap;
import java.util.Map;

import org.objectweb.asm.Label;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ErrorBranchInstrumenter {

	protected static final Logger logger = LoggerFactory.getLogger(ErrorBranchInstrumenter.class);
	
	protected ErrorConditionMethodAdapter mv;
	
	protected String methodName;
	
	public ErrorBranchInstrumenter(ErrorConditionMethodAdapter mv) {
		this.mv = mv;
		this.methodName = mv.getMethodName();
	}
	
	public Map getMethodCallee(String desc) {
		Type[] args = Type.getArgumentTypes(desc);
		Map to = new HashMap();
		for (int i = args.length - 1; i >= 0; i--) {
			int loc = mv.newLocal(args[i]);
			mv.storeLocal(loc);
			to.put(i, loc);
		}

		mv.dup();//callee
		return to;
	}
	
	public void restoreMethodParameters(Map to, String desc) {
		Type[] args = Type.getArgumentTypes(desc);

		for (int i = 0; i < args.length; i++) {
			mv.loadLocal(to.get(i));
		}	
	}

	public void visitMethodInsn(int opcode, String owner, String name, String desc) {
		throw new RuntimeException("This method should not be called since ASM5 API is used");
	}

	
	public void visitMethodInsn(int opcode, String owner, String name, String desc, boolean itf) {
		
	}

	public void visitFieldInsn(int opcode, String owner, String name, String desc) {
		
	}
	
	public void visitTypeInsn(int opcode, String type) {
				
	}
	
	public void visitInsn(int opcode) {

	}
	
	public void visitIntInsn(int opcode,
            int operand) {	
	}

	
	protected void insertBranch(int opcode, String exception) {
		Label origTarget = new Label();
		mv.tagBranch();
		mv.visitJumpInsn(opcode, origTarget);
		mv.visitTypeInsn(Opcodes.NEW, exception);
		mv.visitInsn(Opcodes.DUP);
		mv.visitMethodInsn(Opcodes.INVOKESPECIAL, exception,
		                      "", "()V", false);
		mv.visitInsn(Opcodes.ATHROW);
		mv.visitLabel(origTarget);
		mv.tagBranchExit();
	}

	protected void insertBranchWithoutTag(int opcode, String exception) {
		Label origTarget = new Label();
		mv.visitJumpInsn(opcode, origTarget);
		mv.visitTypeInsn(Opcodes.NEW, exception);
		mv.visitInsn(Opcodes.DUP);
		mv.visitMethodInsn(Opcodes.INVOKESPECIAL, exception,
		                      "", "()V", false);
		mv.visitInsn(Opcodes.ATHROW);
		mv.visitLabel(origTarget);
	}
	
	protected void tagBranchStart() {
		mv.tagBranch();
	}

	protected void tagBranchEnd() {
		mv.tagBranchExit();
	}

}