Extended test for ARM thumb relocations and added reference disassembly

tests/test_thumb_addend.py

@@ -0,0 +1,65 @@
+import pelfy._main as _main
+import os
+
+def test_thumb_addend_extraction():
+    # Path to the test object file
+    obj_path = os.path.join('tests', 'obj', 'stencils_armv7thumb_O3_THM_MOVW.o')
+    elf = _main.open_elf_file(obj_path)
+
+    # Collect all relocations of interest
+    reloc_addends: list[tuple[str, int, int, str]] = []
+    for reloc in elf.get_relocations():
+        if reloc.type.startswith('R_ARM_THM'):
+            reloc_addends.append((reloc.type, reloc['r_offset'], reloc['r_addend'], reloc.symbol.name))
+
+    # Reference values from the .asm file (addend = 0 for all Thumb relocations)
+    reference = [
+        ('R_ARM_THM_MOVW_ABS_NC', None, 0, 'dummy_int'),
+        ('R_ARM_THM_MOVT_ABS', None, 0, 'dummy_int'),
+        ('R_ARM_THM_MOVW_ABS_NC', None, 0, 'dummy_float'),
+        ('R_ARM_THM_MOVT_ABS', None, 0, 'dummy_float'),
+        ('R_ARM_THM_JUMP24', 0x14, 0, 'auxsub_get_42'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'result_int'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_float_int'),
+        ('R_ARM_THM_JUMP24', 0xC, 0, 'result_float_float'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_int_int'),
+        ('R_ARM_THM_JUMP24', 0xC, 0, 'result_int_float'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'aux_get_42'),
+        ('R_ARM_THM_JUMP24', 0x12, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2, 0, 'aux_get_42'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_float'),
+        ('R_ARM_THM_JUMP24', 0x2, 0, 'result_int'),
+        ('R_ARM_THM_JUMP24', 0x4, 0, 'result_float'),
+        ('R_ARM_THM_JUMP24', 0x28, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2C, 0, 'sqrtf'),
+        ('R_ARM_THM_JUMP24', 0x20, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x24, 0, 'sqrtf'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'expf'),
+        ('R_ARM_THM_JUMP24', 0x12, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2, 0, 'expf'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'logf'),
+        ('R_ARM_THM_JUMP24', 0x12, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2, 0, 'logf'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'sinf'),
+        ('R_ARM_THM_JUMP24', 0x12, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2, 0, 'sinf'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'cosf'),
+        ('R_ARM_THM_JUMP24', 0x12, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2, 0, 'cosf'),
+        ('R_ARM_THM_JUMP24', 0xA, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0xA, 0, 'tanf'),
+        ('R_ARM_THM_JUMP24', 0x12, 0, 'result_float'),
+        ('R_ARM_THM_CALL', 0x2, 0, 'tanf'),
+    ]
+    # For each reference, check that at least one matching relocation has the expected addend
+    for ref_type, _, ref_addend, ref_symbol in reference:
+        found = False
+        addend = 0
+        for typ, offset, addend, symbol in reloc_addends:
+            if typ == ref_type and symbol == ref_symbol and addend == ref_addend:
+                found = True
+                break
+        assert found, f"Missing or incorrect addend for {ref_type} {ref_symbol} (value={addend:X}, expected {ref_addend})"