Source code for claripy.backends.backend_smtlib_solvers.z3_popen

import subprocess
import logging

import re
from . import SMTLibSolverBackend, PopenSolverProxy
from ...errors import MissingSolverError

log = logging.getLogger(__name__)


def get_version():
    try:
        version_string = subprocess.check_output(["z3", "-version"]).decode("utf-8")
        version_match = re.match("Z3 version (.*)\n", version_string)

        if not version_match:
            return False, None, f"Found malformed version string: {version_string}"

        return True, version_match.group(1), None

    except subprocess.CalledProcessError as ex:
        return False, None, f"Not found, error: {ex}"
    except OSError as ex:
        return False, None, f"Not found, error: {ex}"


IS_INSTALLED, VERSION, ERROR = get_version()


class Z3Proxy(PopenSolverProxy):
    def __init__(self, timeout=None, max_memory=None):
        self.timeout = timeout
        self.max_memory = max_memory
        self.installed = False
        p = None
        super().__init__(p)

    def create_process(self):
        if not IS_INSTALLED:
            raise MissingSolverError("Z3 not found! Please install Z3 before using this backend")
        cmd = ["z3", "-smt2", "-in"]
        if self.timeout is not None:
            cmd.append(f"-t:{self.timeout//1000}")  # our timeout is in milliseconds
        if self.max_memory is not None:
            cmd.append(f"-memory:{self.max_memory}")

        p = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        self.installed = True
        return p


class SolverBackendZ3(SMTLibSolverBackend):
    def solver(self, timeout=None, max_memory=None):
        """
        This function should return an instance of whatever object handles
        solving for this backend. For example, in Z3, this would be z3.Solver().
        """
        return Z3Proxy(timeout=timeout, max_memory=max_memory)


from ... import backend_manager as backend_manager

backend_manager.backends._register_backend(SolverBackendZ3(), "smtlib_z3", False, False)