test_multi_open_file.py

import angr
import os

import logging
l = logging.getLogger('angr.tests.test_multi_open_file')

test_location = os.path.dirname(os.path.realpath(__file__))

def test_multi_open_file():
    test_bin = os.path.join(test_location, "..", "..", "binaries", "tests", "x86_64", "test_multi_open_file")
    # auto_load_libs cannot be disabled as the test fails
    b = angr.Project(test_bin) 

    pg = b.factory.simulation_manager()
    pg.active[0].options.discard("LAZY_SOLVES")
    pg.explore()

    assert len(pg.deadended) == 1

    # See the source file in binaries/tests_src/test_multi_open_file.c
    # for the tests run
    for p in pg.deadended:
        assert p.posix.dumps(2) == b""

        # Check that the temp file was deleted
        assert p.fs._files == {}

        # Check that the deleted temp file contained the appropriate string
        for event in p.history.events:
            if event.type == 'fs_unlink':
                simfile = p.fs.unlinks[event.objects['unlink_idx']][1]
                assert simfile.concretize() == b'foobar and baz'
                break
        else:
            assert False


if __name__ == "__main__":
    test_multi_open_file()