soot.jimple.infoflow.collections.strategies.containers.TestConstantStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of soot-infoflow Show documentation

Soot extending data flow tracking components for Java

The newest version!

package soot.jimple.infoflow.collections.strategies.containers;

import java.util.concurrent.ConcurrentHashMap;

import soot.Local;
import soot.SootMethod;
import soot.Value;
import soot.jimple.Stmt;
import soot.jimple.infoflow.InfoflowManager;
import soot.jimple.infoflow.collections.analyses.ListSizeAnalysis;
import soot.jimple.infoflow.collections.context.IntervalContext;
import soot.jimple.infoflow.collections.context.UnknownContext;
import soot.jimple.infoflow.collections.strategies.containers.shift.IShiftOperation;
import soot.jimple.infoflow.data.ContainerContext;

/**
 * Strategy that reasons about maps with constant keys and lists with constant
 * indices. Uses {@link ListSizeAnalysis} to retrieve the list size, thus,
 * should only be used for testing the semantics of list operations.
 *
 * @author Tim Lange
 */
public class TestConstantStrategy extends AbstractListStrategy {
	private final ConcurrentHashMap implicitIndices;

	public TestConstantStrategy(InfoflowManager manager) {
		super(manager);
		this.implicitIndices = new ConcurrentHashMap<>();
	}

	public TestConstantStrategy(InfoflowManager manager, IShiftOperation shiftOp) {
		super(manager, shiftOp);
		this.implicitIndices = new ConcurrentHashMap<>();
	}

	@Override
	public ContainerContext getNextPosition(Value value, Stmt stmt) {
		if (!shouldResolveIndex(value))
			return UnknownContext.v();

		return getContextFromImplicitKey(value, stmt, false);
	}

	@Override
	public ContainerContext getLastPosition(Value value, Stmt stmt) {
		if (!shouldResolveIndex(value))
			return UnknownContext.v();

		return getContextFromImplicitKey(value, stmt, true);
	}

	private ContainerContext getContextFromImplicitKey(Value value, Stmt stmt, boolean decr) {
		if (value instanceof Local) {
			SootMethod currMethod = manager.getICFG().getMethodOf(stmt);
			ListSizeAnalysis lstSizeAnalysis = implicitIndices.computeIfAbsent(currMethod,
					sm -> new ListSizeAnalysis(manager.getICFG().getOrCreateUnitGraph(sm)));
			ListSizeAnalysis.ListSize size = lstSizeAnalysis.getFlowBefore(stmt).get(value);
			if (size != null && !size.isBottom()) {
				return new IntervalContext(decr ? size.getSize() - 1 : size.getSize());
			}
		}

		return UnknownContext.v();
	}
}