Object file injection, first part

[centaur.git] / src / model / insert.c

diff --git a/src/model/insert.c b/src/model/insert.c
index d4e19d3a0dd5d0629afe599ea076eea4c31e859b..312be26c79393d2723e15a115feb37071d7206c7 100644 (file)
--- a/src/model/insert.c
+++ b/src/model/insert.c
@@ -1,6 +1,6 @@
 #include <assert.h>
 #include <sys/types.h>
-#include <gelf.h>
+#include <libelf/gelf.h>
 #include <libelfu/libelfu.h>
 
 
@@ -20,7 +20,7 @@
  * the same address in memory, and everything BEFORE it is shifted to
  * lower addresses, making space for the new data in-between.
  */
-GElf_Xword elfu_mInsertBefore(ElfuElf *me, GElf_Off off, GElf_Xword size)
+GElf_Xword elfu_mInsertSpaceBefore(ElfuElf *me, GElf_Off off, GElf_Xword size)
 {
   ElfuScn *ms;
   ElfuPhdr *mp;
@@ -121,8 +121,12 @@ GElf_Xword elfu_mInsertBefore(ElfuElf *me, GElf_Off off, GElf_Xword size)
  *
  * PHDRs will be patched such that everything AFTER off is shifted to
  * higher addresses, making space for the new data in-between.
+ *
+ * CAUTION: This also moves NOBITS sections. If such are present,
+ *          use mExpandNobits() first and then inject at the end of
+ *          the expansion site.
  */
-GElf_Xword elfu_mInsertAfter(ElfuElf *me, GElf_Off off, GElf_Xword size)
+GElf_Xword elfu_mInsertSpaceAfter(ElfuElf *me, GElf_Off off, GElf_Xword size)
 {
   ElfuScn *ms;
   ElfuPhdr *mp;
@@ -207,3 +211,74 @@ GElf_Xword elfu_mInsertAfter(ElfuElf *me, GElf_Off off, GElf_Xword size)
 
   return size;
 }
+
+
+
+
+
+/* Update cross-references */
+static void shiftSections(ElfuElf *me, ElfuScn *first)
+{
+  ElfuScn *ms = first;
+  size_t firstIndex = elfu_mScnIndex(me, first);
+
+  do {
+    if (ms == me->shstrtab) {
+      me->ehdr.e_shstrndx++;
+    }
+
+    ms = CIRCLEQ_LOOP_NEXT(&me->scnList, ms, elem);
+  } while (ms != CIRCLEQ_FIRST(&me->scnList));
+
+  CIRCLEQ_FOREACH(ms, &me->scnList, elem) {
+    switch (ms->shdr.sh_type) {
+      case SHT_REL:
+      case SHT_RELA:
+        if (ms->shdr.sh_info >= firstIndex) {
+          ms->shdr.sh_info++;
+        }
+      case SHT_DYNAMIC:
+      case SHT_HASH:
+      case SHT_SYMTAB:
+      case SHT_DYNSYM:
+      case SHT_GNU_versym:
+      case SHT_GNU_verdef:
+      case SHT_GNU_verneed:
+        if (ms->shdr.sh_link >= firstIndex) {
+          ms->shdr.sh_link++;
+        }
+    }
+  }
+}
+
+
+/*
+ * Insert a section into an ELF model, /before/ a given other section
+ */
+void elfu_mInsertScnInChainBefore(ElfuElf *me, ElfuScn *oldscn, ElfuScn *newscn)
+{
+  assert(me);
+  assert(oldscn);
+  assert(newscn);
+
+  shiftSections(me, oldscn);
+
+  CIRCLEQ_INSERT_BEFORE(&me->scnList, oldscn, newscn, elem);
+}
+
+
+/*
+ * Insert a section into an ELF model, /after/ a given other section
+ */
+void elfu_mInsertScnInChainAfter(ElfuElf *me, ElfuScn *oldscn, ElfuScn *newscn)
+{
+  assert(me);
+  assert(oldscn);
+  assert(newscn);
+
+  if (oldscn != CIRCLEQ_LAST(&me->scnList)) {
+    shiftSections(me, CIRCLEQ_NEXT(oldscn, elem));
+  }
+
+  CIRCLEQ_INSERT_AFTER(&me->scnList, oldscn, newscn, elem);
+}