de.hdu.pvs.crashfinder.instrument.EveryStmtInstrumenter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crashFinder Show documentation
A tool for localizing the software regression errors
The newest version!
package de.hdu.pvs.crashfinder.instrument;

import java.io.Writer;
import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.Set;

import com.ibm.wala.shrikeBT.ConstantInstruction;
import com.ibm.wala.shrikeBT.Constants;
import com.ibm.wala.shrikeBT.Disassembler;
import com.ibm.wala.shrikeBT.IInstruction;
import com.ibm.wala.shrikeBT.Instruction;
import com.ibm.wala.shrikeBT.MethodData;
import com.ibm.wala.shrikeBT.MethodEditor;
import com.ibm.wala.shrikeBT.Util;
import com.ibm.wala.shrikeBT.analysis.Verifier;
import com.ibm.wala.shrikeBT.shrikeCT.ClassInstrumenter;
import com.ibm.wala.shrikeCT.ClassReader;
import com.ibm.wala.shrikeCT.ClassWriter;

import de.hdu.pvs.crashfinder.util.Globals;
import de.hdu.pvs.crashfinder.util.Utils;
import de.hdu.pvs.crashfinder.util.WALAUtils;

public class EveryStmtInstrumenter extends AbstractInstrumenter {

	static final String fieldName = "_Stmt_enable_trace";

	static final Instruction getTracer = Util.makeGet(StmtTracer.class,
			"tracer");
	static final Instruction callTrace = Util.makeInvoke(StmtTracer.class,
			"trace", new Class[] { String.class });

	private Set skippedClasses = null;

	private Set instrumentedClasses = null;

	public EveryStmtInstrumenter() {
		// empty on purpose
	}

	// full name
	public void setSkippedClasses(Collection classes) {
		skippedClasses = new LinkedHashSet();
		skippedClasses.addAll(classes);
	}

	public void setInstrumentedClassPrefix(Collection classes) {
		instrumentedClasses = new LinkedHashSet();
		instrumentedClasses.addAll(classes);
	}

	@Override
	protected void doClass(ClassInstrumenter ci, Writer w) throws Exception {
		final String className = ci.getReader().getName();
		w.write("Class: " + className + "\n");
		w.flush();

		if (this.skippedClasses != null) {
			String cName = Utils.translateSlashToDot(className);
			if (Utils.startWith(cName,
					this.skippedClasses.toArray(new String[0]))) {
				// if(this.skippedClasses.contains(cName)) {
				System.out.print("Skip. " + cName + "\n");
				return;
			}
		}

		if (this.instrumentedClasses != null) {
			String cName = Utils.translateSlashToDot(className);
			if (!Utils.startWith(cName,
					this.instrumentedClasses.toArray(new String[0]))) {
				System.out.print("Not in instrumented class: " + cName + "\n");
				return;
			}
		}

		for (int m = 0; m < ci.getReader().getMethodCount(); m++) {
			MethodData d = ci.visitMethod(m);
			// d could be null, e.g., if the method is abstract or native
			if (d != null) {
				w.write("Instrumenting " + ci.getReader().getMethodName(m)
						+ " " + ci.getReader().getMethodType(m) + ":\n");
				w.flush();
				if (disasm) {
					w.write("Initial ShrikeBT code:\n");
					(new Disassembler(d)).disassembleTo(w);
					w.flush();
				}
				if (verify) {
					Verifier v = new Verifier(d);
					v.verify();
				}

				MethodEditor me = new MethodEditor(d);
				me.beginPass();

				// the unique method signature
				String methodSig = WALAUtils.getMethodSignature(d);

				// instrument every instruction
				int length = me.getInstructions().length;
				for (int i = 0; i < length; i++) {
					IInstruction inst = me.getInstructions()[i];
					String instStr = inst.toString();
					// FIXME do sanitize, remove SEP and line break
					final String msg = methodSig + SEP + instStr + SEP + i; // a
																			// unique
																			// encoding
					me.insertBefore(i, new MethodEditor.Patch() {
						@Override
						public void emitTo(MethodEditor.Output w) {
							// w.emit(getSysErr);
							w.emit(getTracer);
							w.emit(ConstantInstruction.makeString(msg));
							w.emit(callTrace);
							// keep the statistics
							InstrumentStats.addInsertedInstructions(1);
						}
					});
				}

				// this updates the data, instrumentation ends
				me.applyPatches();
				if (disasm) {
					w.write("Final ShrikeBT code:\n");
					(new Disassembler(d)).disassembleTo(w);
					w.write(Globals.lineSep);
					w.flush();
				}
			}
		}

		if (ci.isChanged()) {
			ClassWriter cw = ci.emitClass();
			cw.addField(ClassReader.ACC_PUBLIC | ClassReader.ACC_STATIC,
					fieldName, Constants.TYPE_boolean,
					new ClassWriter.Element[0]);
			instrumenter.outputModifiedClass(ci, cw);
		}
	}

}