#include "api/z3.h"
#include "util/debug.h"
#include <iostream>
void test_print(Z3_context ctx, Z3_ast_vector av) {
Z3_set_ast_print_mode(ctx, Z3_PRINT_SMTLIB2_COMPLIANT);
Z3_ast* args = new Z3_ast[Z3_ast_vector_size(ctx, av)];
for (unsigned i = 0; i < Z3_ast_vector_size(ctx, av); ++i) {
args[i] = Z3_ast_vector_get(ctx, av, i);
}
Z3_ast a = Z3_mk_and(ctx, Z3_ast_vector_size(ctx, av), args);
Z3_inc_ref(ctx, a);
delete[] args;
char const* spec1 = Z3_benchmark_to_smtlib_string(ctx, "test", nullptr, nullptr, nullptr, 0, nullptr, a);
Z3_dec_ref(ctx, a);
std::cout << "spec1: benchmark->string\n" << spec1 << "\n";
std::cout << "attempting to parse spec1...\n";
Z3_ast_vector b =
Z3_parse_smtlib2_string(ctx,
spec1,
0,
nullptr,
nullptr,
0,
nullptr,
nullptr);
std::cout << "parse successful, converting ast->string\n";
Z3_ast_vector_inc_ref(ctx, b);
char const* spec2 = Z3_ast_vector_to_string(ctx, b);
std::cout << "spec2: string->ast->string\n" << spec2 << "\n";
Z3_ast_vector_dec_ref(ctx, b);
}
void test_parseprint(char const* spec) {
Z3_context ctx = Z3_mk_context(nullptr);
std::cout << "spec:\n" << spec << "\n";
Z3_ast_vector a =
Z3_parse_smtlib2_string(ctx,
spec,
0,
nullptr,
nullptr,
0,
nullptr,
nullptr);
std::cout << "done parsing\n";
Z3_ast_vector_inc_ref(ctx, a);
test_print(ctx, a);
std::cout << "done printing\n";
Z3_ast_vector_dec_ref(ctx, a);
Z3_del_context(ctx);
}
static bool is_error = false;
void setError(Z3_context c, Z3_error_code e) {
is_error = true;
}
void test_eval(Z3_context ctx, Z3_string spec, bool shouldFail) {
std::cout << "spec:\n" << spec << "\n";
std::string resp;
is_error = false;
resp = Z3_eval_smtlib2_string(ctx, spec);
if (!is_error)
std::cout << "response:\n" << resp << "\n";
if (shouldFail != is_error) {
if (shouldFail)
throw std::runtime_error("should have failed");
else
throw std::runtime_error("should have succeeded");
}
}
void test_repeated_eval() {
char const* spec1 =
"(push)\n"
"(declare-datatypes (T) ((list (nil) (cons (car T) (cdr list)))))\n"
"(declare-const x Int)\n"
"(declare-const l (list Int))\n"
"(declare-fun f ((list Int)) Bool)\n"
"(assert (f (cons x l)))\n"
"(check-sat)\n"
"(pop)\n";
char const* spec2 =
"(push)\n"
"(declare-const a (Array Int Int))\n"
"(declare-const b (Array (Array Int Int) Bool))\n"
"(assert (select b a))\n"
"(assert (= b ((as const (Array (Array Int Int) Bool)) true)))\n"
"(assert (= b (store b a true)))\n"
"(declare-const b1 (Array Bool Bool))\n"
"(declare-const b2 (Array Bool Bool))\n"
"(assert (= ((as const (Array Bool Bool)) false) ((_ map and) b1 b2)))\n"
"(check-sat)\n"
"(pop)\n";
char const* spec3 =
"(push)\n"
"(declare-const a@ (Array Int Int))\n"
"(declare-const b (Array (Array Int Int) Bool))\n"
"(assert (select b a))\n"
"(check-sat)\n"
"(pop)\n";
char const* spec4 =
"(push)\n"
"(declare-const a (Array Int Int))\n"
"(declare-const b# (Array (Array Int Int) Bool))\n"
"(assert (select b a))\n"
"(check-sat)\n"
"(pop)\n";
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
std::cout << "testing Z3_eval_smtlib2_string\n";
try {
test_eval(ctx, spec1, false);
std::cout << "successful call after successful call\n";
test_eval(ctx, spec2, false);
std::cout << "failing call after successful call\n";
test_eval(ctx, spec3, true);
std::cout << "failing call after failing call\n";
test_eval(ctx, spec4, true);
std::cout << "successful call after failing call\n";
test_eval(ctx, spec1, false);
}
catch(...) {
std::cout << "Error: uncaught exception\n";
throw;
}
std::cout << "done evaluating\n";
Z3_del_context(ctx);
}
void test_ho_curried_application() {
char const* spec =
"(set-logic HO_ALL)\n"
"(declare-fun transfer () (-> (-> Int Bool) (-> Int Bool)))\n"
"(assert (forall ((P (-> Int Bool))) (=> (P 0) ((transfer P) 0))))\n"
"(declare-fun top () (-> Int Bool))\n"
"(assert (forall ((x Int)) (top x)))\n"
"(assert (not ((transfer top) 0)))\n"
"(check-sat)\n";
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
test_eval(ctx, spec, false);
Z3_del_context(ctx);
}
void test_ho_choice_expression() {
char const* spec =
"(set-logic HO_ALL)\n"
"(declare-sort U 0)\n"
"(declare-fun P () (-> U Bool))\n"
"(assert (exists ((x U)) (P x)))\n"
"(declare-fun witness () U)\n"
"(assert (= witness (choice ((x U)) (P x))))\n"
"(assert (not (P witness)))\n"
"(check-sat)\n";
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
test_eval(ctx, spec, false);
Z3_del_context(ctx);
}
void test_name(Z3_string spec, Z3_string expected_name) {
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
std::cout << "spec:\n" << spec << "\n";
is_error = false;
Z3_ast_vector a =
Z3_parse_smtlib2_string(ctx,
spec,
0,
nullptr,
nullptr,
0,
nullptr,
nullptr);
std::cout << "done parsing\n";
ENSURE(is_error == (expected_name == nullptr));
if (is_error) {
Z3_del_context(ctx);
return;
}
Z3_ast_vector_inc_ref(ctx, a);
ENSURE(Z3_ast_vector_size(ctx, a) == 1)
Z3_ast c = Z3_ast_vector_get(ctx, a, 0);
Z3_inc_ref(ctx, c);
Z3_app app = Z3_to_app(ctx, c);
Z3_func_decl decl = Z3_get_app_decl(ctx, app);
Z3_symbol symbol = Z3_get_decl_name(ctx, decl);
Z3_string name = Z3_get_symbol_string(ctx, symbol);
bool success = std::string(name) == std::string(expected_name);
Z3_dec_ref(ctx, c);
Z3_ast_vector_dec_ref(ctx, a);
Z3_del_context(ctx);
ENSURE(success);
}
void test_symbol_escape() {
#define SYMBOL_ASSERTION(N) \
"(declare-const " N " Bool)\n" \
"(assert " N ")\n" \
"(check-sat)\n"
std::cout << "testing Z3_eval_smtlib2_string\n";
try {
test_name(SYMBOL_ASSERTION("|a|"), "a");
test_name(SYMBOL_ASSERTION("|a\\|"), nullptr);
test_name(SYMBOL_ASSERTION("|a\\||"), "a|");
test_name(SYMBOL_ASSERTION("|a\\\\|"), "a\\");
test_name(SYMBOL_ASSERTION("|a\\\\||"), nullptr);
test_name(SYMBOL_ASSERTION("|a\\a|"), "a\\a");
test_name(SYMBOL_ASSERTION("|a\\a"), nullptr);
}
catch(...) {
std::cout << "Error: uncaught exception\n";
throw;
}
std::cout << "done evaluating\n";
}
void test_builtin_signature_clash() {
char const* rejected[] = {
"(declare-fun and (Bool Bool) Int)",
"(define-fun not ((a Bool)) Bool false)",
"(declare-const true Bool)",
"(define-fun = ((a Int) (b Int)) Bool true)",
"(define-fun ite ((c Bool) (a Int) (b Int)) Int 0)",
"(define-fun + ((a Int) (b Int)) Int 0)",
"(define-fun-rec and ((a Bool) (b Bool)) Bool false)"
};
for (char const* spec : rejected) {
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
is_error = false;
Z3_parse_smtlib2_string(ctx, spec, 0, nullptr, nullptr, 0, nullptr, nullptr);
ENSURE(is_error);
Z3_del_context(ctx);
}
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
test_eval(ctx,
"(declare-fun and (Int Int) Int)\n"
"(assert (= (and 1 2) 0))\n"
"(check-sat)\n",
false);
Z3_del_context(ctx);
}
void tst_smt2print_parse() {
char const* spec1 =
"(declare-datatypes (T) ((list (nil) (cons (car T) (cdr list)))))\n"
"(declare-const x Int)\n"
"(declare-const l (list Int))\n"
"(declare-fun f ((list Int)) Bool)\n"
"(assert (f (cons x l)))\n";
test_parseprint(spec1);
char const* spec2 =
"(declare-const x Int)\n"
"(declare-const a (Array Int Int))\n"
"(declare-const b (Array (Array Int Int) Bool))\n"
"(assert (select b a))\n"
"(assert (= b ((as const (Array (Array Int Int) Bool)) true)))\n"
"(assert (= b (store b a true)))\n"
"(declare-const b1 (Array Bool Bool))\n"
"(declare-const b2 (Array Bool Bool))\n"
"(assert (= ((as const (Array Bool Bool)) false) ((_ map and) b1 b2)))\n";
test_parseprint(spec2);
char const* spec3 =
"(declare-datatypes () ((list (nil) (cons (car tree) (cdr list))) (tree (leaf) (node (n list)))))\n"
"(declare-const x tree)\n"
"(declare-const l list)\n"
"(declare-fun f (list) Bool)\n"
"(assert (f (cons x l)))\n";
test_parseprint(spec3);
char const* spec4 =
"(declare-const x Real)\n"
"(declare-const y Int)\n"
"(assert (= x 0.0))\n"
"(assert (= y 6))\n"
"(assert (> (/ x 1.4) (to_real y)))";
test_parseprint(spec4);
char const* spec5 =
"(declare-const x (_ BitVec 4))\n"
"(declare-const y (_ BitVec 4))\n"
"(assert (bvule x (bvmul y (concat ((_ extract 2 0) x) ((_ extract 3 3) #xf0)))))";
test_parseprint(spec5);
char const* spec6 =
"(assert (= \"abc\" \"abc\"))";
test_parseprint(spec6);
test_repeated_eval();
test_ho_curried_application();
test_ho_choice_expression();
test_symbol_escape();
test_builtin_signature_clash();
{
char const* spec =
"(set-option :smtlib2_compliant true)\n"
"(set-logic ALL)\n"
"(declare-datatype SBVRational ((SBV.Rational (sbv.rat.numerator Int) (sbv.rat.denominator Int))))\n"
"(define-fun sbv.rat.eq ((x SBVRational) (y SBVRational)) Bool\n"
" (= (* (sbv.rat.numerator x) (sbv.rat.denominator y))\n"
" (* (sbv.rat.denominator x) (sbv.rat.numerator y))))\n"
"(define-fun s4 () Real 0.0)\n"
"(declare-fun s0 () SBVRational)\n"
"(assert (< 0 (sbv.rat.denominator s0)))\n"
"(declare-fun s1 () SBVRational)\n"
"(assert (< 0 (sbv.rat.denominator s1)))\n"
"(define-fun s2 () Int (sbv.rat.denominator s1))\n"
"(define-fun s3 () Real (to_real s2))\n"
"(define-fun s5 () Bool (= s3 s4))\n"
"(define-fun s6 () Int (sbv.rat.numerator s1))\n"
"(define-fun s7 () Real (to_real s6))\n"
"(define-fun s8 () Real (/ s7 s3))\n"
"(define-fun s9 () Real (ite s5 s4 s8))\n"
"(define-fun s10 () Int (sbv.rat.denominator s0))\n"
"(define-fun s11 () Real (to_real s10))\n"
"(define-fun s12 () Bool (= s4 s11))\n"
"(define-fun s13 () Int (sbv.rat.numerator s0))\n"
"(define-fun s14 () Real (to_real s13))\n"
"(define-fun s15 () Real (/ s14 s11))\n"
"(define-fun s16 () Real (ite s12 s4 s15))\n"
"(define-fun s17 () Bool (= s9 s16))\n"
"(define-fun s18 () Bool (sbv.rat.eq s0 s1))\n"
"(assert s17)\n"
"(assert (not s18))\n"
"(check-sat)\n";
Z3_context ctx = Z3_mk_context(nullptr);
Z3_set_error_handler(ctx, setError);
is_error = false;
std::string resp = Z3_eval_smtlib2_string(ctx, spec);
Z3_del_context(ctx);
std::cout << "Issue #10166 response: " << resp << "\n";
ENSURE(!is_error);
ENSURE(resp.find("unsat") != std::string::npos);
ENSURE(resp.find("unknown") == std::string::npos);
}
}