def test_simplify_dead_assign_1():
# if a register is used ever, it should not be simplified away
arch = archinfo.arch_from_id('AMD64')
block = ailment.Block(0x1337, 10)
n = count(start=1)
important = 0x999
block.statements.extend([
ailment.Assignment(
next(n),
ailment.Register(next(n), None, arch.registers['rdi'][0], 64),
ailment.Const(next(n), None, 0x13371337, 64),
ins_addr=0x1337,
), # rdi = 0x13371337
ailment.Stmt.Call(
important,
ailment.Const(next(n), None, 0x400080, 64),
ins_addr=0x1338,
), # Call(0x400080), which uses rdi but also overwrites rdi (since it is a caller-saved argument)
])
b = block_simplify(block)
nose.tools.assert_equal(len(b.statements), 2)
nose.tools.assert_equal(b.statements[0].idx, 1)
nose.tools.assert_equal(b.statements[1].idx, important)
def test_simplify_dead_assign_0():
block = ailment.Block(0x1337, 10)
n = count()
important = 0x999
block.statements.extend(
[
ailment.Assignment(
next(n),
ailment.Register(next(n), None, 1, 64),
ailment.Const(next(n), None, 100, 64),
ins_addr=0x1337,
),
ailment.Assignment(
important,
ailment.Register(next(n), None, 1, 64),
ailment.Const(next(n), None, 101, 64),
ins_addr=0x1338,
),
ailment.Stmt.Jump(
next(n),
ailment.Expr.Const(None, None, 0x3333, 64),
ins_addr=0x1338
),
]
)
b = block_simplify(block)
nose.tools.assert_equal(len(b.statements), 2)
nose.tools.assert_equal(b.statements[0].idx, important)
def test_simplify_pointless_assign():
arch = archinfo.arch_from_id('AMD64')
block = ailment.Block(0x1337, 10)
block.statements.append(
ailment.Assignment(
0,
ailment.Register(1, None, arch.registers['rax'][0], 64),
ailment.Register(2, None, arch.registers['rax'][0], 64),
ins_addr=0x1337,
))
block.statements.append(
ailment.Assignment(
3,
ailment.Register(4, None, arch.registers['rbx'][0], 64),
ailment.Register(5, None, arch.registers['rcx'][0], 64),
ins_addr=0x1338,
))
b = block_simplify(block)
nose.tools.assert_equal(len(b.statements), 1)
nose.tools.assert_equal(b.statements[0].idx, 3)
def test_can_be_instantiated_with_an_ailment_block():
block = ailment.Block(0x42, original_size=4)
subject = Subject(block, None)
nose.tools.assert_equals(subject.content, block)
nose.tools.assert_equals(subject.type, SubjectType.Block)
def test_can_be_instantiated_with_an_ailment_block(self):
block = ailment.Block(0x42, original_size=4)
subject = Subject(block)
assert subject.content == block
assert subject.type == SubjectType.Block
