symbolic_execution_math_func_example.py

import angr
import claripy
import logging

logging.getLogger('angr').disabled = True
logging.getLogger('angr').propagate = False
logging.getLogger('cle').disabled = True
logging.getLogger('cle').propagate = False

from expression.components import *
from code_generation.c_code_generation import CCodeGenerator
from symbolic_execution.symbolic_expression_extraction import SymbolicExpressionExtractor
from code_generation.bin_code_generation import CFile

def main():
    c = Const(4)
    v1 = Var('a', "double")
    v2 = Var('b', "double")
    expr = PowFunc(v1, AddOp(CosFunc(v2), c))
    do_expr(expr, "double", simplified=False)


def do_expr(expr, ret_type, simplified=True):
    print("Natural Expression:")
    print(expr)
    ccg = CCodeGenerator(expr, ret_type=ret_type)
    generated_c_code = ccg.generate_code()
    code = generated_c_code.code
    print("C code:")
    print(code)
    target_func = generated_c_code.wrapper_func
    var_names = generated_c_code.expr_var_names
    var_ctypes = generated_c_code.expr_var_ctypes
    
    c_file_name = 'example_c.c'
    cfile = CFile(c_file_name, code)
    bin_file_name = cfile.compile()


    print("Symbolic Expression:")
    see  = SymbolicExpressionExtractor(bin_file_name)
    extracted_symexpr = see.extract(target_func, var_names, var_ctypes, ret_type, simplified)
    ast = extracted_symexpr.symex_expr
    print(ast)




if __name__ == "__main__":
    main()