def simp_cc_conds(expr_simp, expr):
if (expr.is_op("CC_U>=") and test_cc_eq_args(expr, "FLAG_SUB_CF")):
expr = ExprCond(ExprOp(TOK_INF_UNSIGNED, *expr.args[0].args),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_U<") and test_cc_eq_args(expr, "FLAG_SUB_CF")):
expr = ExprOp(TOK_INF_UNSIGNED, *expr.args[0].args)
elif (expr.is_op("CC_NEG") and test_cc_eq_args(expr, "FLAG_SIGN_SUB")):
expr = ExprOp(TOK_INF_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_POS") and test_cc_eq_args(expr, "FLAG_SIGN_SUB")):
expr = ExprCond(ExprOp(TOK_INF_SIGNED, *expr.args[0].args),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_EQ") and test_cc_eq_args(expr, "FLAG_EQ")):
arg = expr.args[0].args[0]
expr = ExprOp(TOK_EQUAL, arg, ExprInt(0, arg.size))
elif (expr.is_op("CC_NE") and test_cc_eq_args(expr, "FLAG_EQ")):
arg = expr.args[0].args[0]
expr = ExprCond(ExprOp(TOK_EQUAL, arg, ExprInt(0, arg.size)),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_NE") and test_cc_eq_args(expr, "FLAG_EQ_CMP")):
expr = ExprCond(ExprOp(TOK_EQUAL, *expr.args[0].args), ExprInt(0, 1),
ExprInt(1, 1))
elif (expr.is_op("CC_EQ") and test_cc_eq_args(expr, "FLAG_EQ_CMP")):
expr = ExprOp(TOK_EQUAL, *expr.args[0].args)
elif (expr.is_op("CC_NE") and test_cc_eq_args(expr, "FLAG_EQ_AND")):
expr = ExprOp("&", *expr.args[0].args)
elif (expr.is_op("CC_EQ") and test_cc_eq_args(expr, "FLAG_EQ_AND")):
expr = ExprCond(ExprOp("&", *expr.args[0].args), ExprInt(0, 1),
ExprInt(1, 1))
elif (expr.is_op("CC_S>") and test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF",
"FLAG_EQ_CMP",
)):
expr = ExprCond(ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_S>") and len(expr.args) == 3
and expr.args[0].is_op("FLAG_SIGN_SUB")
and expr.args[2].is_op("FLAG_EQ_CMP")
and expr.args[0].args == expr.args[2].args
and expr.args[1].is_int(0)):
expr = ExprCond(ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_S>=")
and test_cc_eq_args(expr, "FLAG_SIGN_SUB", "FLAG_SUB_OF")):
expr = ExprCond(ExprOp(TOK_INF_SIGNED, *expr.args[0].args),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_S<")
and test_cc_eq_args(expr, "FLAG_SIGN_SUB", "FLAG_SUB_OF")):
expr = ExprOp(TOK_INF_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_S<=") and test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF",
"FLAG_EQ_CMP",
)):
expr = ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_S<=") and len(expr.args) == 3
and expr.args[0].is_op("FLAG_SIGN_SUB")
and expr.args[2].is_op("FLAG_EQ_CMP")
and expr.args[0].args == expr.args[2].args
and expr.args[1].is_int(0)):
expr = ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_U<=") and test_cc_eq_args(
expr,
"FLAG_SUB_CF",
"FLAG_EQ_CMP",
)):
expr = ExprOp(TOK_INF_EQUAL_UNSIGNED, *expr.args[0].args)
elif (expr.is_op("CC_U>") and test_cc_eq_args(
expr,
"FLAG_SUB_CF",
"FLAG_EQ_CMP",
)):
expr = ExprCond(ExprOp(TOK_INF_EQUAL_UNSIGNED, *expr.args[0].args),
ExprInt(0, 1), ExprInt(1, 1))
elif (expr.is_op("CC_S<")
and test_cc_eq_args(expr, "FLAG_SIGN_ADD", "FLAG_ADD_OF")):
arg0, arg1 = expr.args[0].args
expr = ExprOp(TOK_INF_SIGNED, arg0, -arg1)
return expr
def simp_cc_conds(_, expr):
"""
High level simplifications. Example:
CC_U<(FLAG_SUB_CF(A, B) => A <u B
"""
if (expr.is_op("CC_U>=") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF"
)):
expr = ExprCond(
ExprOp(TOK_INF_UNSIGNED, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size))
elif (expr.is_op("CC_U<") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF"
)):
expr = ExprOp(TOK_INF_UNSIGNED, *expr.args[0].args)
elif (expr.is_op("CC_NEG") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB"
)):
expr = ExprOp(TOK_INF_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_POS") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB"
)):
expr = ExprCond(
ExprOp(TOK_INF_SIGNED, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_EQ") and
test_cc_eq_args(
expr,
"FLAG_EQ"
)):
arg = expr.args[0].args[0]
expr = ExprOp(TOK_EQUAL, arg, ExprInt(0, arg.size))
elif (expr.is_op("CC_NE") and
test_cc_eq_args(
expr,
"FLAG_EQ"
)):
arg = expr.args[0].args[0]
expr = ExprCond(
ExprOp(TOK_EQUAL,arg, ExprInt(0, arg.size)),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_NE") and
test_cc_eq_args(
expr,
"FLAG_EQ_CMP"
)):
expr = ExprCond(
ExprOp(TOK_EQUAL, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_EQ") and
test_cc_eq_args(
expr,
"FLAG_EQ_CMP"
)):
expr = ExprOp(TOK_EQUAL, *expr.args[0].args)
elif (expr.is_op("CC_NE") and
test_cc_eq_args(
expr,
"FLAG_EQ_AND"
)):
expr = ExprOp("&", *expr.args[0].args)
elif (expr.is_op("CC_EQ") and
test_cc_eq_args(
expr,
"FLAG_EQ_AND"
)):
expr = ExprCond(
ExprOp("&", *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_S>") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF",
"FLAG_EQ_CMP",
)):
expr = ExprCond(
ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_S>") and
len(expr.args) == 3 and
expr.args[0].is_op("FLAG_SIGN_SUB") and
expr.args[2].is_op("FLAG_EQ_CMP") and
expr.args[0].args == expr.args[2].args and
expr.args[1].is_int(0)):
expr = ExprCond(
ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_S>=") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF"
)):
expr = ExprCond(
ExprOp(TOK_INF_SIGNED, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_S<") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF"
)):
expr = ExprOp(TOK_INF_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_S<=") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF",
"FLAG_EQ_CMP",
)):
expr = ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_S<=") and
len(expr.args) == 3 and
expr.args[0].is_op("FLAG_SIGN_SUB") and
expr.args[2].is_op("FLAG_EQ_CMP") and
expr.args[0].args == expr.args[2].args and
expr.args[1].is_int(0)):
expr = ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_U<=") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF",
"FLAG_EQ_CMP",
)):
expr = ExprOp(TOK_INF_EQUAL_UNSIGNED, *expr.args[0].args)
elif (expr.is_op("CC_U>") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF",
"FLAG_EQ_CMP",
)):
expr = ExprCond(
ExprOp(TOK_INF_EQUAL_UNSIGNED, *expr.args[0].args),
ExprInt(0, expr.size),
ExprInt(1, expr.size)
)
elif (expr.is_op("CC_S<") and
test_cc_eq_args(
expr,
"FLAG_SIGN_ADD",
"FLAG_ADD_OF"
)):
arg0, arg1 = expr.args[0].args
expr = ExprOp(TOK_INF_SIGNED, arg0, -arg1)
return expr
def simp_cc_conds(expr_simp, expr):
if (expr.is_op("CC_U>=") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF"
)):
expr = ExprCond(
ExprOp(TOK_INF_UNSIGNED, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1))
elif (expr.is_op("CC_U<") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF"
)):
expr = ExprOp(TOK_INF_UNSIGNED, *expr.args[0].args)
elif (expr.is_op("CC_NEG") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB"
)):
expr = ExprOp(TOK_INF_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_POS") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB"
)):
expr = ExprCond(
ExprOp(TOK_INF_SIGNED, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_EQ") and
test_cc_eq_args(
expr,
"FLAG_EQ"
)):
arg = expr.args[0].args[0]
expr = ExprOp(TOK_EQUAL, arg, ExprInt(0, arg.size))
elif (expr.is_op("CC_NE") and
test_cc_eq_args(
expr,
"FLAG_EQ"
)):
arg = expr.args[0].args[0]
expr = ExprCond(
ExprOp(TOK_EQUAL,arg, ExprInt(0, arg.size)),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_NE") and
test_cc_eq_args(
expr,
"FLAG_EQ_CMP"
)):
expr = ExprCond(
ExprOp(TOK_EQUAL, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_EQ") and
test_cc_eq_args(
expr,
"FLAG_EQ_CMP"
)):
expr = ExprOp(TOK_EQUAL, *expr.args[0].args)
elif (expr.is_op("CC_NE") and
test_cc_eq_args(
expr,
"FLAG_EQ_AND"
)):
expr = ExprOp("&", *expr.args[0].args)
elif (expr.is_op("CC_EQ") and
test_cc_eq_args(
expr,
"FLAG_EQ_AND"
)):
expr = ExprCond(
ExprOp("&", *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_S>") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF",
"FLAG_EQ_CMP",
)):
expr = ExprCond(
ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_S>") and
len(expr.args) == 3 and
expr.args[0].is_op("FLAG_SIGN_SUB") and
expr.args[2].is_op("FLAG_EQ_CMP") and
expr.args[0].args == expr.args[2].args and
expr.args[1].is_int(0)):
expr = ExprCond(
ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_S>=") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF"
)):
expr = ExprCond(
ExprOp(TOK_INF_SIGNED, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_S<") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF"
)):
expr = ExprOp(TOK_INF_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_S<=") and
test_cc_eq_args(
expr,
"FLAG_SIGN_SUB",
"FLAG_SUB_OF",
"FLAG_EQ_CMP",
)):
expr = ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_S<=") and
len(expr.args) == 3 and
expr.args[0].is_op("FLAG_SIGN_SUB") and
expr.args[2].is_op("FLAG_EQ_CMP") and
expr.args[0].args == expr.args[2].args and
expr.args[1].is_int(0)):
expr = ExprOp(TOK_INF_EQUAL_SIGNED, *expr.args[0].args)
elif (expr.is_op("CC_U<=") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF",
"FLAG_EQ_CMP",
)):
expr = ExprOp(TOK_INF_EQUAL_UNSIGNED, *expr.args[0].args)
elif (expr.is_op("CC_U>") and
test_cc_eq_args(
expr,
"FLAG_SUB_CF",
"FLAG_EQ_CMP",
)):
expr = ExprCond(
ExprOp(TOK_INF_EQUAL_UNSIGNED, *expr.args[0].args),
ExprInt(0, 1),
ExprInt(1, 1)
)
elif (expr.is_op("CC_S<") and
test_cc_eq_args(
expr,
"FLAG_SIGN_ADD",
"FLAG_ADD_OF"
)):
arg0, arg1 = expr.args[0].args
expr = ExprOp(TOK_INF_SIGNED, arg0, -arg1)
return expr