Implement mInsertAfter, for post-.data injection

[centaur.git] / src / main.c

diff --git a/src/main.c b/src/main.c
index 991fe09423b20247cfc0a4765873f3707934d3d9..1e06d2998bca0642143bbb2deee1a51aaab14ed5 100644 (file)
--- a/src/main.c
+++ b/src/main.c
@@ -67,12 +67,20 @@ int main(int argc, char **argv)
 
     if (me) {
       printf("Model successfully loaded.\n");
-
       elfu_mCheck(me);
-      printf("Model checked.\n");
+      printf("Input model checked.\n");
 
-      elfu_mToElf(me, hOut.e);
+      if (opts.insertBeforeSz) {
+        elfu_mInsertBefore(me, opts.insertBeforeOffs, opts.insertBeforeSz);
+      }
+
+      if (opts.insertAfterSz) {
+        elfu_mInsertAfter(me, opts.insertAfterOffs, opts.insertAfterSz);
+      }
 
+      elfu_mCheck(me);
+      printf("Output model checked.\n");
+      elfu_mToElf(me, hOut.e);
       printf("Model converted to ELF, ready to be written.\n");
     } else {
       printf("Failed to load model.\n");