Examples with SmtCheckSatQuery org.evosuite.symbolic.solver.smt.SmtCheckSatQuery used on opensource projects

Example 1 with SmtCheckSatQuery

use of org.evosuite.symbolic.solver.smt.SmtCheckSatQuery in project evosuite by EvoSuite.

the class Z3Str2Solver method solve.

@Override
public SolverResult solve(Collection<Constraint<?>> constraints) throws SolverTimeoutException, IOException, SolverParseException, SolverEmptyQueryException, SolverErrorException {
    SmtCheckSatQuery smtCheckSatQuery = buildSmtQuerty(constraints);
    if (smtCheckSatQuery.getConstantDeclarations().isEmpty()) {
        logger.debug("Z3-str2 input has no variables");
        throw new SolverEmptyQueryException("Z3-str2 input has no variables");
    }
    if (smtCheckSatQuery.getAssertions().isEmpty()) {
        Map<String, Object> emptySolution = new HashMap<String, Object>();
        SolverResult emptySAT = SolverResult.newSAT(emptySolution);
        return emptySAT;
    }
    Z3Str2QueryPrinter printer = new Z3Str2QueryPrinter();
    String smtQueryStr = printer.print(smtCheckSatQuery);
    logger.debug("Z3-str2 input:");
    logger.debug(smtQueryStr);
    int timeout = (int) Properties.DSE_CONSTRAINT_SOLVER_TIMEOUT_MILLIS;
    File tempDir = createNewTmpDir();
    String z3TempFileName = tempDir.getAbsolutePath() + File.separatorChar + EVOSUITE_Z3_STR_FILENAME;
    if (Properties.Z3_STR2_PATH == null) {
        String errMsg = "Property Z3_STR_PATH should be setted in order to use the Z3StrSolver!";
        logger.error(errMsg);
        throw new IllegalStateException(errMsg);
    }
    try {
        FileIOUtils.writeFile(smtQueryStr, z3TempFileName);
        String z3Cmd = Properties.Z3_STR2_PATH + " -f " + z3TempFileName;
        ByteArrayOutputStream stdout = new ByteArrayOutputStream();
        launchNewProcess(z3Cmd, smtQueryStr, timeout, stdout);
        String z3str2ResultStr = stdout.toString("UTF-8");
        Z3Str2ResultParser parser = new Z3Str2ResultParser();
        Set<Variable<?>> variables = getVariables(constraints);
        Map<String, Object> initialValues = getConcreteValues(variables);
        SolverResult solverResult;
        if (addMissingVariables()) {
            solverResult = parser.parse(z3str2ResultStr, initialValues);
        } else {
            solverResult = parser.parse(z3str2ResultStr);
        }
        if (solverResult.isSAT()) {
            // check if solution is correct, otherwise return UNSAT
            boolean check = checkSAT(constraints, solverResult);
            if (!check) {
                logger.debug("Z3-str2 solution does not solve the constraint system!");
                SolverResult unsatResult = SolverResult.newUNSAT();
                return unsatResult;
            }
        }
        return solverResult;
    } catch (UnsupportedEncodingException e) {
        throw new EvosuiteError("UTF-8 should not cause this exception!");
    } finally {
        File tempFile = new File(z3TempFileName);
        if (tempFile.exists()) {
            tempFile.delete();
        }
    }
}

Example 2 with SmtCheckSatQuery

use of org.evosuite.symbolic.solver.smt.SmtCheckSatQuery in project evosuite by EvoSuite.

the class CVC4Solver method buildSmtCheckSatQuery.

private static SmtCheckSatQuery buildSmtCheckSatQuery(Collection<Constraint<?>> constraints) {
    ConstraintToCVC4Visitor v = new ConstraintToCVC4Visitor(true);
    SmtVariableCollector varCollector = new SmtVariableCollector();
    SmtOperatorCollector funCollector = new SmtOperatorCollector();
    List<SmtAssertion> smtAssertions = new LinkedList<SmtAssertion>();
    for (Constraint<?> c : constraints) {
        SmtExpr smtExpr = c.accept(v, null);
        if (smtExpr != null) {
            SmtAssertion smtAssertion = new SmtAssertion(smtExpr);
            smtAssertions.add(smtAssertion);
            smtExpr.accept(varCollector, null);
            smtExpr.accept(funCollector, null);
        }
    }
    Set<SmtVariable> variables = varCollector.getSmtVariables();
    if (variables.isEmpty()) {
        // no variables, constraint system is trivial
        return null;
    }
    List<SmtFunctionDefinition> functionDefinitions = new LinkedList<SmtFunctionDefinition>();
    final boolean addCharToInt = funCollector.getOperators().contains(Operator.CHAR_TO_INT);
    if (addCharToInt) {
        String charToIntFunction = buildCharToIntFunction();
        SmtFunctionDefinition funcDefinition = new SmtFunctionDefinition(charToIntFunction);
        functionDefinitions.add(funcDefinition);
    }
    final boolean addIntToChar = funCollector.getOperators().contains(Operator.INT_TO_CHAR);
    if (addIntToChar) {
        String intToCharFunction = buildIntToCharFunction();
        SmtFunctionDefinition funcDefinition = new SmtFunctionDefinition(intToCharFunction);
        functionDefinitions.add(funcDefinition);
    }
    List<SmtFunctionDeclaration> functionDeclarations = new LinkedList<SmtFunctionDeclaration>();
    for (SmtVariable var : variables) {
        String varName = var.getName();
        if (var instanceof SmtIntVariable) {
            SmtFunctionDeclaration intVar = SmtExprBuilder.mkIntFunctionDeclaration(varName);
            functionDeclarations.add(intVar);
        } else if (var instanceof SmtRealVariable) {
            SmtFunctionDeclaration realVar = SmtExprBuilder.mkRealFunctionDeclaration(varName);
            functionDeclarations.add(realVar);
        } else if (var instanceof SmtStringVariable) {
            SmtFunctionDeclaration stringVar = SmtExprBuilder.mkStringFunctionDeclaration(varName);
            functionDeclarations.add(stringVar);
        } else {
            throw new RuntimeException("Unknown variable type " + var.getClass().getCanonicalName());
        }
    }
    SmtCheckSatQuery smtQuery = new SmtCheckSatQuery(new LinkedList<SmtConstantDeclaration>(), functionDeclarations, functionDefinitions, smtAssertions);
    return smtQuery;
}

Example 3 with SmtCheckSatQuery

use of org.evosuite.symbolic.solver.smt.SmtCheckSatQuery in project evosuite by EvoSuite.

the class Z3Solver method solve.

@Override
public SolverResult solve(Collection<Constraint<?>> constraints) throws SolverTimeoutException, IOException, SolverParseException, SolverEmptyQueryException, SolverErrorException {
    long hard_timeout = Properties.DSE_CONSTRAINT_SOLVER_TIMEOUT_MILLIS;
    Set<Variable<?>> variables = new HashSet<Variable<?>>();
    for (Constraint<?> c : constraints) {
        Set<Variable<?>> c_variables = c.getVariables();
        variables.addAll(c_variables);
    }
    SmtCheckSatQuery smtCheckSatQuery = buildSmtQuery(constraints, variables);
    if (smtCheckSatQuery.getConstantDeclarations().isEmpty()) {
        logger.debug("Z3 SMT query has no variables");
        throw new SolverEmptyQueryException("Z3 SMT query has no variables");
    }
    if (smtCheckSatQuery.getAssertions().isEmpty()) {
        Map<String, Object> emptySolution = new HashMap<String, Object>();
        SolverResult emptySAT = SolverResult.newSAT(emptySolution);
        return emptySAT;
    }
    Z3QueryPrinter printer = new Z3QueryPrinter();
    String smtQueryStr = printer.print(smtCheckSatQuery, hard_timeout);
    logger.debug("Z3 Query:");
    logger.debug(smtQueryStr);
    if (Properties.Z3_PATH == null) {
        String errMsg = "Property Z3_PATH should be setted in order to use the Z3 Solver!";
        logger.error(errMsg);
        throw new IllegalStateException(errMsg);
    }
    String z3Cmd = Properties.Z3_PATH + " -smt2 -in";
    ByteArrayOutputStream stdout = new ByteArrayOutputStream();
    launchNewProcess(z3Cmd, smtQueryStr, (int) hard_timeout, stdout);
    String z3ResultStr = stdout.toString("UTF-8");
    Map<String, Object> initialValues = getConcreteValues(variables);
    Z3ResultParser resultParser;
    if (this.addMissingVariables()) {
        resultParser = new Z3ResultParser(initialValues);
    } else {
        resultParser = new Z3ResultParser();
    }
    SolverResult result = resultParser.parseResult(z3ResultStr);
    return result;
}

Example 4 with SmtCheckSatQuery

use of org.evosuite.symbolic.solver.smt.SmtCheckSatQuery in project evosuite by EvoSuite.

the class Z3Solver method buildSmtQuery.

private static SmtCheckSatQuery buildSmtQuery(Collection<Constraint<?>> constraints, Set<Variable<?>> variables) {
    List<SmtConstantDeclaration> constantDeclarations = new LinkedList<SmtConstantDeclaration>();
    for (Variable<?> v : variables) {
        String varName = v.getName();
        if (v instanceof IntegerVariable) {
            SmtConstantDeclaration intVar = SmtExprBuilder.mkIntConstantDeclaration(varName);
            constantDeclarations.add(intVar);
        } else if (v instanceof RealVariable) {
            SmtConstantDeclaration realVar = SmtExprBuilder.mkRealConstantDeclaration(varName);
            constantDeclarations.add(realVar);
        } else if (v instanceof StringVariable) {
        // ignore string variables
        } else {
            throw new RuntimeException("Unknown variable type " + v.getClass().getCanonicalName());
        }
    }
    List<SmtAssertion> assertions = new LinkedList<SmtAssertion>();
    for (Constraint<?> c : constraints) {
        ConstraintToZ3Visitor v = new ConstraintToZ3Visitor();
        SmtExpr bool_expr = c.accept(v, null);
        if (bool_expr != null && bool_expr.isSymbolic()) {
            SmtAssertion newAssertion = new SmtAssertion(bool_expr);
            assertions.add(newAssertion);
        }
    }
    SmtCheckSatQuery smtCheckSatQuery = new SmtCheckSatQuery(constantDeclarations, assertions);
    return smtCheckSatQuery;
}

Example 5 with SmtCheckSatQuery

use of org.evosuite.symbolic.solver.smt.SmtCheckSatQuery in project evosuite by EvoSuite.

the class Z3Str2Solver method buildSmtQuerty.

private static SmtCheckSatQuery buildSmtQuerty(Collection<Constraint<?>> constraints) {
    ConstraintToZ3Str2Visitor v = new ConstraintToZ3Str2Visitor();
    List<SmtAssertion> assertions = new LinkedList<SmtAssertion>();
    SmtVariableCollector varCollector = new SmtVariableCollector();
    SmtOperatorCollector opCollector = new SmtOperatorCollector();
    for (Constraint<?> c : constraints) {
        SmtExpr smtExpr = c.accept(v, null);
        if (smtExpr != null) {
            SmtAssertion newAssertion = new SmtAssertion(smtExpr);
            assertions.add(newAssertion);
            smtExpr.accept(varCollector, null);
            smtExpr.accept(opCollector, null);
        }
    }
    Set<SmtVariable> smtVariables = varCollector.getSmtVariables();
    Set<Operator> smtOperators = opCollector.getOperators();
    boolean addCharToIntFunction;
    if (smtOperators.contains(SmtOperation.Operator.CHAR_TO_INT)) {
        addCharToIntFunction = true;
    } else {
        addCharToIntFunction = false;
    }
    Set<SmtVariable> smtVariablesToDeclare = new HashSet<SmtVariable>(smtVariables);
    if (addCharToIntFunction) {
        Set<SmtStringVariable> charVariables = buildCharVariables();
        smtVariablesToDeclare.addAll(charVariables);
    }
    List<SmtConstantDeclaration> constantDeclarations = new LinkedList<SmtConstantDeclaration>();
    for (SmtVariable v1 : smtVariablesToDeclare) {
        String varName = v1.getName();
        if (v1 instanceof SmtIntVariable) {
            SmtConstantDeclaration constantDecl = SmtExprBuilder.mkIntConstantDeclaration(varName);
            constantDeclarations.add(constantDecl);
        } else if (v1 instanceof SmtRealVariable) {
            SmtConstantDeclaration constantDecl = SmtExprBuilder.mkRealConstantDeclaration(varName);
            constantDeclarations.add(constantDecl);
        } else if (v1 instanceof SmtStringVariable) {
            SmtConstantDeclaration constantDecl = SmtExprBuilder.mkStringConstantDeclaration(varName);
            constantDeclarations.add(constantDecl);
        } else {
            throw new RuntimeException("Unknown variable type " + v1.getClass().getCanonicalName());
        }
    }
    List<SmtFunctionDefinition> functionDefinitions = new LinkedList<SmtFunctionDefinition>();
    if (addCharToIntFunction) {
        String charToInt = buildCharToIntFunction();
        SmtFunctionDefinition newFunctionDef = new SmtFunctionDefinition(charToInt);
        functionDefinitions.add(newFunctionDef);
    }
    SmtCheckSatQuery smtCheckSatQuery = new SmtCheckSatQuery(constantDeclarations, functionDefinitions, assertions);
    return smtCheckSatQuery;
}