crypto.constraints.ValueConstraint 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.constraints;

import crypto.analysis.errors.ConstraintError;
import crypto.extractparameter.CallSiteWithExtractedValue;
import crysl.rule.CrySLObject;
import crysl.rule.CrySLSplitter;
import crysl.rule.CrySLValueConstraint;
import crysl.rule.ISLConstraint;
import java.util.AbstractMap;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class ValueConstraint extends EvaluableConstraint {

    protected ValueConstraint(ISLConstraint origin, ConstraintSolver context) {
        super(origin, context);
    }

    @Override
    public void evaluate() {
        CrySLValueConstraint valCons = (CrySLValueConstraint) origin;
        List> values =
                getValFromVar(valCons.getVar(), valCons);
        if (values.isEmpty()) {
            return;
        }

        List lowerCaseValues =
                valCons.getValueRange().parallelStream().map(String::toLowerCase).toList();
        for (Map.Entry val : values) {
            if (!lowerCaseValues.contains(val.getKey().toLowerCase())) {
                ConstraintError error =
                        new ConstraintError(
                                context.getSeed(),
                                val.getValue(),
                                context.getSpecification(),
                                valCons);
                errors.add(error);
            }
        }
    }

    private List> getValFromVar(
            CrySLObject var, ISLConstraint cons) {
        final String varName = var.getVarName();
        final Map valueCollection =
                extractValueAsString(varName);

        List> values = new ArrayList<>();
        if (couldNotExtractValues(valueCollection, cons)) {
            return values;
        }

        for (Map.Entry e : valueCollection.entrySet()) {
            CrySLSplitter splitter = var.getSplitter();
            final CallSiteWithExtractedValue location = e.getValue();
            String val = e.getKey();
            if (splitter != null) {
                int ind = splitter.getIndex();
                String splitElement = splitter.getSplitter();
                if (ind > 0) {
                    String[] splits = val.split(splitElement);
                    if (splits.length > ind) {
                        values.add(new AbstractMap.SimpleEntry<>(splits[ind], location));
                    } else {
                        values.add(new AbstractMap.SimpleEntry<>("", location));
                    }
                } else {
                    values.add(
                            new AbstractMap.SimpleEntry<>(val.split(splitElement)[ind], location));
                }
            } else {
                values.add(new AbstractMap.SimpleEntry<>(val, location));
            }
        }
        return values;
    }
}