mirror of
https://github.com/angr/angr
synced 2026-08-17 12:23:11 -04:00
f939c5b88c
* Enable ruff isort rule * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
59 lines
1.7 KiB
Python
Executable file
59 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# pylint:disable=missing-class-docstring,no-self-use
|
|
from __future__ import annotations
|
|
|
|
import math
|
|
from unittest import TestCase, main
|
|
|
|
import claripy
|
|
|
|
import angr
|
|
|
|
|
|
class TestSqrt(TestCase):
|
|
def test_sqrt_symbolic(self):
|
|
ins_bytes = b"\xf3\x0f\x51\xc9" # sqrtss xmm1, xmm1
|
|
proj = angr.load_shellcode(ins_bytes, "amd64", load_address=0)
|
|
state = proj.factory.blank_state(addr=0)
|
|
|
|
xmm1 = claripy.FPS("v", claripy.FSORT_FLOAT)
|
|
state.regs.xmm1 = xmm1.to_bv()
|
|
simgr = proj.factory.simgr(state)
|
|
|
|
simgr.step(num_inst=1)
|
|
assert len(simgr.active) == 1
|
|
|
|
final = simgr.active[0]
|
|
result = final.regs.xmm1[31:0].raw_to_fp()
|
|
|
|
a = int(final.solver.eval(xmm1, extra_constraints=(result == 2.0,)))
|
|
assert a == 4
|
|
b = int(final.solver.eval(xmm1, extra_constraints=(result == 4.0,)))
|
|
assert b == 16
|
|
|
|
def test_sqrt_concrete(self):
|
|
ins_bytes = b"\xf3\x0f\x51\xc9" # sqrtss xmm1, xmm1
|
|
proj = angr.load_shellcode(ins_bytes, "amd64", load_address=0)
|
|
state = proj.factory.blank_state(addr=0)
|
|
|
|
xmm1 = claripy.FPV(200000, claripy.FSORT_FLOAT)
|
|
state.regs.xmm1 = xmm1.to_bv()
|
|
simgr = proj.factory.simgr(state)
|
|
|
|
simgr.step(num_inst=1)
|
|
assert len(simgr.active) == 1
|
|
|
|
final = simgr.active[0]
|
|
result = final.regs.xmm1[31:0].raw_to_fp()
|
|
|
|
a = final.solver.eval(result)
|
|
assert abs(a - math.sqrt(200000)) < 0.001
|
|
|
|
def test_sqrt_concrete_eager_evaluation(self):
|
|
a = claripy.FPV(2.0, claripy.FSORT_DOUBLE)
|
|
b = claripy.fpSqrt(a)
|
|
assert abs(b.concrete_value - 1.414) < 0.001
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|