crypto.typestate.LabeledMatcherTransition Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of CryptoAnalysis Show documentation

CogniCrypt_SAST: CrySL-to-Static Analysis Compiler

The newest version!

package crypto.typestate;

import boomerang.scene.DeclaredMethod;
import crypto.utils.MatcherUtils;
import crysl.rule.CrySLMethod;
import java.util.Collection;
import java.util.Objects;
import java.util.Optional;
import typestate.finiteautomata.MatcherTransition;
import typestate.finiteautomata.State;

public class LabeledMatcherTransition extends MatcherTransition {

    private final Collection methods;

    public LabeledMatcherTransition(State from, Collection methods, State to) {
        super(from, "", Parameter.This, to, Type.OnCallOrOnCallToReturn);

        this.methods = methods;
    }

    @Override
    public boolean matches(DeclaredMethod declaredMethod) {
        return getMatching(declaredMethod).isPresent();
    }

    /**
     * Return the {@link CrySLMethod}'s that match the given method. As the method is taken from a
     * statement, we need to apply the matching logic defined here, to get the {@link CrySLMethod}s
     * that were resolved to the matching {@link DeclaredMethod}s.
     *
     * @param declaredMethod the given method
     * @return The {@link CrySLMethod}'s matching the given declared method.
     */
    public Optional getMatching(DeclaredMethod declaredMethod) {
        for (CrySLMethod method : methods) {
            if (MatcherUtils.matchCryslMethodAndDeclaredMethod(method, declaredMethod)) {
                return Optional.of(method);
            }
        }

        return Optional.empty();
    }

    public Collection getMethods() {
        return methods;
    }

    @Override
    public String toString() {
        return super.toString();
    }

    @Override
    public boolean equals(Object obj) {
        return super.equals(obj)
                && obj instanceof LabeledMatcherTransition other
                && Objects.equals(methods, other.getMethods());
    }

    @Override
    public int hashCode() {
        return Objects.hash(super.hashCode(), from(), to(), methods);
    }
}