diff --git a/test/test_files/Makefile b/test/test_files/Makefile index 751b4b052415167549bcd7e245674bae6c1cd897..2c2ab5f20b2ae0e68e3fbcf42913da0312ae347a 100644 --- a/test/test_files/Makefile +++ b/test/test_files/Makefile @@ -1,5 +1,5 @@ CC=gcc -SHARED_FLAGS=-shared -fPIC -O3 +SHARED_FLAGS=-shared -fPIC -O3 -m64 CFLAGS=$(SHARED_FLAGS) -Wl,-Bsymbolic-functions RPATH_DIR=rpath_dir RPATH_SUBDIR=$(RPATH_DIR)/rpath_subdir diff --git a/test/test_library.py b/test/test_library.py index 76aeac385cd506d11ddfb6c4db453e7692da678c..2c60d67c59f14c4edfe39e0f070e5b3947e1a1de 100644 --- a/test/test_library.py +++ b/test/test_library.py @@ -425,5 +425,17 @@ class TestLibrary(unittest.TestCase): self.assertIn(binary.fullname, lib.get_users_by_name('external_caller')) self.assertIn(binary.fullname, lib.get_users_by_name('second_level_caller')) + def test_7_search_for_plt(self): + _, library = create_store_and_lib(TEST_LIBRARY) + searched_plt = library._search_for_plt() + section_plt = library._elffile.get_section_by_name('.plt') + self.assertEquals(searched_plt['sh_offset'], section_plt['sh_offset']) + + def test_7_dynamic_rela_plt(self): + _, library = create_store_and_lib(TEST_LIBRARY) + searched_plt = library._create_mock_rela_plt() + section_plt = library._elffile.get_section_by_name('.rela.plt') + self.assertEquals(searched_plt['sh_offset'], section_plt['sh_offset']) + if __name__ == '__main__': unittest.main()