src/main.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3
   4 #include <getopt.h>
   5 #include <libelf/libelf.h>
   6 #include <libelf/gelf.h>
   7
   8 #include <libelfu/libelfu.h>
   9
  10 #include "elfhandle.h"
  11 #include "options.h"
  12 #include "printing.h"
  13
  14
  15 int main(int argc, char **argv)
  16 {
  17   CLIOpts opts = { 0 };
  18   ELFHandles hIn = { 0 };
  19   ELFHandles hOut = { 0 };
  20   int exitval = EXIT_SUCCESS;
  21   ElfuElf *me;
  22
  23   /* Is libelf alive and well? */
  24   if (elf_version(EV_CURRENT) == EV_NONE) {
  25     fprintf(stderr, "libelf init error: %s\n", elf_errmsg(-1));
  26   }
  27
  28
  29   /* Parse and validate user input */
  30   parseOptions(&opts, argc, argv);
  31
  32
  33   /* Open input/output files */
  34   openElf(&hIn, opts.fnInput, ELF_C_READ);
  35   if (!hIn.e) {
  36     exitval = EXIT_FAILURE;
  37     goto EXIT;
  38   }
  39
  40
  41   /* Now that we have a (hopefully) sane environment, execute commands.
  42    * Printing will have to be reimplemented based on the memory model.
  43    */
  44   if (opts.printHeader) {
  45     printHeader(hIn.e);
  46   }
  47
  48   if (opts.printSegments) {
  49     printSegments(hIn.e);
  50   }
  51
  52   if (opts.printSections) {
  53     printSections(hIn.e);
  54   }
  55
  56
  57   me = elfu_mFromElf(hIn.e);
  58   if (me) {
  59     closeElf(&hIn);
  60     printf("Model successfully loaded.\n");
  61
  62     elfu_mDumpElf(me);
  63
  64     elfu_mCheck(me);
  65     printf("Input model checked.\n");
  66   } else {
  67     printf("Failed to load model, aborting.\n");
  68     goto EXIT;
  69   }
  70
  71
  72   /* Copy the input ELF to the output file if the latter is specified.
  73    * Perform requested transformations on the memory model on-the-fly. */
  74   if (!opts.fnOutput) {
  75     printf("No output file specified - no further operations performed.\n");
  76   } else {
  77     if (opts.expandNobitsOffs) {
  78       elfu_mExpandNobits(me, opts.expandNobitsOffs);
  79     }
  80
  81     if (opts.insertBeforeSz) {
  82       elfu_mInsertSpaceBefore(me, opts.insertBeforeOffs, opts.insertBeforeSz);
  83     }
  84
  85     if (opts.insertAfterSz) {
  86       elfu_mInsertSpaceAfter(me, opts.insertAfterOffs, opts.insertAfterSz);
  87     }
  88
  89     if (opts.fnReladd) {
  90       ELFHandles hRel = { 0 };
  91       ElfuElf *mrel = NULL;
  92
  93       openElf(&hRel, opts.fnReladd, ELF_C_READ);
  94       if (!hRel.e) {
  95         printf("--reladd: Failed to open file for --reladd, skipping operation.\n");
  96       } else {
  97         mrel = elfu_mFromElf(hRel.e);
  98         closeElf(&hRel);
  99         if (!me) {
 100           printf("--reladd: Failed to load model for --reladd, skipping operation.\n");
 101         } else {
 102           printf("--reladd: Model successfully loaded.\n");
 103           elfu_mCheck(mrel);
 104           printf("--reladd: Input model checked.\n");
 105           elfu_mReladd(me, mrel);
 106         }
 107       }
 108
 109     }
 110
 111     elfu_mCheck(me);
 112     printf("Output model checked.\n");
 113
 114
 115     openElf(&hOut, opts.fnOutput, ELF_C_WRITE);
 116     if (!hOut.e) {
 117       printf("Failed to open output file. Aborting.\n");
 118       exitval = EXIT_FAILURE;
 119       goto EXIT;
 120     }
 121
 122     elfu_mToElf(me, hOut.e);
 123     printf("Model converted to ELF, ready to be written.\n");
 124   }
 125
 126
 127
 128 EXIT:
 129   if (hOut.e) {
 130     if (elf_update(hOut.e, ELF_C_WRITE) < 0) {
 131       fprintf(stderr, "elf_update() failed: %s\n", elf_errmsg(-1));
 132     }
 133     closeElf(&hOut);
 134   }
 135
 136   if (hIn.e) {
 137     closeElf(&hIn);
 138   }
 139
 140   return (exitval);
 141 }