Petr Machata 3eda79e
@@ -, +, @@ 
Petr Machata 3eda79e
 relocation
Petr Machata 3eda79e
- In general they are.  But IRELATIVE relocations are sorted to come
Petr Machata 3eda79e
  last, and PLT entries are not sorted accordingly.
Petr Machata 3eda79e
---
Petr Machata 3eda79e
 sysdeps/linux-gnu/x86/arch.h |   11 +++++
Petr Machata 3eda79e
 sysdeps/linux-gnu/x86/plt.c  |  101 +++++++++++++++++++++++++++++++++++++++++-
Petr Machata 3eda79e
 2 files changed, 111 insertions(+), 1 deletions(-)
Petr Machata 3eda79e
--- a/sysdeps/linux-gnu/x86/arch.h	
Petr Machata 3eda79e
+++ a/sysdeps/linux-gnu/x86/arch.h	
Petr Machata 3eda79e
@@ -19,6 +19,10 @@ 
Petr Machata 3eda79e
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
Petr Machata 3eda79e
  * 02110-1301 USA
Petr Machata 3eda79e
  */
Petr Machata 3eda79e
+#ifndef LTRACE_X86_ARCH_H
Petr Machata 3eda79e
+#define LTRACE_X86_ARCH_H
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+#include "vect.h"
Petr Machata 3eda79e
 
Petr Machata 3eda79e
 #define BREAKPOINT_VALUE {0xcc}
Petr Machata 3eda79e
 #define BREAKPOINT_LENGTH 1
Petr Machata 3eda79e
@@ -30,9 +34,16 @@ 
Petr Machata 3eda79e
 
Petr Machata 3eda79e
 #define ARCH_HAVE_ADD_PLT_ENTRY
Petr Machata 3eda79e
 
Petr Machata 3eda79e
+#define ARCH_HAVE_LTELF_DATA
Petr Machata 3eda79e
+struct arch_ltelf_data {
Petr Machata 3eda79e
+	struct vect plt_map;
Petr Machata 3eda79e
+};
Petr Machata 3eda79e
+
Petr Machata 3eda79e
 #ifdef __x86_64__
Petr Machata 3eda79e
 #define LT_ELFCLASS	ELFCLASS64
Petr Machata 3eda79e
 #define LT_ELF_MACHINE	EM_X86_64
Petr Machata 3eda79e
 #endif
Petr Machata 3eda79e
 #define LT_ELFCLASS2	ELFCLASS32
Petr Machata 3eda79e
 #define LT_ELF_MACHINE2	EM_386
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+#endif /* LTRACE_X86_ARCH_H */
Petr Machata 3eda79e
--- a/sysdeps/linux-gnu/x86/plt.c	
Petr Machata 3eda79e
+++ a/sysdeps/linux-gnu/x86/plt.c	
Petr Machata 3eda79e
@@ -27,10 +27,19 @@ 
Petr Machata 3eda79e
 #include "library.h"
Petr Machata 3eda79e
 #include "trace.h"
Petr Machata 3eda79e
 
Petr Machata 3eda79e
+static GElf_Addr
Petr Machata 3eda79e
+x86_plt_offset(uint32_t i)
Petr Machata 3eda79e
+{
Petr Machata 3eda79e
+	/* Skip the first PLT entry, which contains a stub to call the
Petr Machata 3eda79e
+	 * resolver.  */
Petr Machata 3eda79e
+	return (i + 1) * 16;
Petr Machata 3eda79e
+}
Petr Machata 3eda79e
+
Petr Machata 3eda79e
 GElf_Addr
Petr Machata 3eda79e
 arch_plt_sym_val(struct ltelf *lte, size_t ndx, GElf_Rela *rela)
Petr Machata 3eda79e
 {
Petr Machata 3eda79e
-	return lte->plt_addr + (ndx + 1) * 16;
Petr Machata 3eda79e
+	uint32_t i = *VECT_ELEMENT(&lte->arch.plt_map, uint32_t, ndx);
Petr Machata 3eda79e
+	return x86_plt_offset(i) + lte->plt_addr;
Petr Machata 3eda79e
 }
Petr Machata 3eda79e
 
Petr Machata 3eda79e
 void *
Petr Machata 3eda79e
@@ -62,3 +71,93 @@ arch_elf_add_plt_entry(struct process *proc, struct ltelf *lte,
Petr Machata 3eda79e
 
Petr Machata 3eda79e
 	return PLT_DEFAULT;
Petr Machata 3eda79e
 }
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+int
Petr Machata 3eda79e
+arch_elf_init(struct ltelf *lte, struct library *lib)
Petr Machata 3eda79e
+{
Petr Machata 3eda79e
+	VECT_INIT(&lte->arch.plt_map, unsigned int);
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+	/* IRELATIVE slots may make the whole situation a fair deal
Petr Machata 3eda79e
+	 * more complex.  On x86{,_64}, the PLT slots are not
Petr Machata 3eda79e
+	 * presented in the order of the corresponding relocations,
Petr Machata 3eda79e
+	 * but in the order it which these symbols are in the symbol
Petr Machata 3eda79e
+	 * table.  That's static symbol table, which may be stripped
Petr Machata 3eda79e
+	 * off, not dynsym--that doesn't contain IFUNC symbols at all.
Petr Machata 3eda79e
+	 * So we have to decode each PLT entry to figure out what
Petr Machata 3eda79e
+	 * entry it corresponds to.  We need to interpret the PLT
Petr Machata 3eda79e
+	 * table to figure this out.
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+	 * On i386, the PLT entry format is as follows:
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+	 *	8048300:   ff 25 0c a0 04 08       jmp    *0x804a00c
Petr Machata 3eda79e
+	 *	8048306:   68 20 00 00 00          push   $0x20
Petr Machata 3eda79e
+	 *	804830b:   e9 e0 ff ff ff          jmp    80482f0 <_init+0x30>
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+	 * For PIE binaries it is the following:
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+	 *	    410:   ff a3 10 00 00 00       jmp    *0x10(%ebx)
Petr Machata 3eda79e
+	 *	    416:   68 00 00 00 00          push   $0x0
Petr Machata 3eda79e
+	 *	    41b:   e9 d0 ff ff ff          jmp    3f0 <_init+0x30>
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+	 * On x86_64, it is:
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+	 *	 400420:   ff 25 f2 0b 20 00       jmpq   *0x200bf2(%rip)        # 601018 <_GLOBAL_OFFSET_TABLE_+0x18>
Petr Machata 3eda79e
+	 *	 400426:   68 00 00 00 00          pushq  $0x0
Petr Machata 3eda79e
+	 *	 40042b:   e9 e0 ff ff ff          jmpq   400410 <_init+0x18>
Petr Machata 3eda79e
+	 *
Petr Machata 3eda79e
+         * On i386, the argument to push is an offset of relocation to
Petr Machata 3eda79e
+	 * use.  The first PLT slot has an offset of 0x0, the second
Petr Machata 3eda79e
+	 * 0x8, etc.  On x86_64, it's directly the index that we are
Petr Machata 3eda79e
+	 * looking for.
Petr Machata 3eda79e
+	 */
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+	/* Here we scan the PLT table and initialize a map of
Petr Machata 3eda79e
+	 * relocation->slot number in lte->arch.plt_map.  */
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+	size_t i;
Petr Machata 3eda79e
+	for (i = 0; i < vect_size(&lte->plt_relocs); ++i) {
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+		GElf_Addr offset = x86_plt_offset(i);
Petr Machata 3eda79e
+		uint32_t reloc_arg = 0;
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+		uint8_t byte;
Petr Machata 3eda79e
+		if (elf_read_next_u8(lte->plt_data, &offset, &byte) < 0
Petr Machata 3eda79e
+		    || byte != 0xff
Petr Machata 3eda79e
+		    || elf_read_next_u8(lte->plt_data, &offset, &byte) < 0
Petr Machata 3eda79e
+		    || (byte != 0xa3 && byte != 0x25))
Petr Machata 3eda79e
+			goto next;
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+		/* Skip immediate argument in the instruction.  */
Petr Machata 3eda79e
+		offset += 4;
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+		if (elf_read_next_u8(lte->plt_data, &offset, &byte) < 0
Petr Machata 3eda79e
+		    || byte != 0x68
Petr Machata 3eda79e
+		    || elf_read_next_u32(lte->plt_data,
Petr Machata 3eda79e
+					 &offset, &reloc_arg) < 0) {
Petr Machata 3eda79e
+			reloc_arg = 0;
Petr Machata 3eda79e
+			goto next;
Petr Machata 3eda79e
+		}
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+		if (lte->ehdr.e_machine == EM_386) {
Petr Machata 3eda79e
+			if (reloc_arg % 8 != 0) {
Petr Machata 3eda79e
+				reloc_arg = 0;
Petr Machata 3eda79e
+				goto next;
Petr Machata 3eda79e
+			}
Petr Machata 3eda79e
+			reloc_arg /= 8;
Petr Machata 3eda79e
+		}
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+	next:
Petr Machata 3eda79e
+		if (VECT_PUSHBACK(&lte->arch.plt_map, &reloc_arg) < 0) {
Petr Machata 3eda79e
+			arch_elf_destroy(lte);
Petr Machata 3eda79e
+			return -1;
Petr Machata 3eda79e
+		}
Petr Machata 3eda79e
+	}
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+	return 0;
Petr Machata 3eda79e
+}
Petr Machata 3eda79e
+
Petr Machata 3eda79e
+void
Petr Machata 3eda79e
+arch_elf_destroy(struct ltelf *lte)
Petr Machata 3eda79e
+{
Petr Machata 3eda79e
+	VECT_DESTROY(&lte->arch.plt_map, uint32_t, NULL, NULL);
Petr Machata 3eda79e
+}
Petr Machata 3eda79e
--