postlinker.cc
上传用户:wstnjxml
上传日期:2014-04-03
资源大小:7248k
文件大小:8k
源码类别:

Windows CE

开发平台:

C/C++

  1. /**********  * Copyright (c) 2004 Greg Parker.  All rights reserved.  *  * Redistribution and use in source and binary forms, with or without  * modification, are permitted provided that the following conditions  * are met:  * 1. Redistributions of source code must retain the above copyright  *    notice, this list of conditions and the following disclaimer.  * 2. Redistributions in binary form must reproduce the above copyright  *    notice, this list of conditions and the following disclaimer in the  *    documentation and/or other materials provided with the distribution.  *  * THIS SOFTWARE IS PROVIDED BY GREG PARKER ``AS IS'' AND ANY EXPRESS OR  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.  **********/ #include <stdio.h> #include <stdlib.h> #include <stdint.h> #include <sys/types.h> #include <sys/stat.h> #include <sys/uio.h> #include <unistd.h> #include <fcntl.h> #include <stdarg.h> #include <string.h> #include "elf.h" #include <map> #include <string> #include <vector> #include <algorithm> using namespace std; #include "got.h" #include "swap.h" #include "image.h" #include "symbol.h" #include "section.h" #include "postlinker.h" #include "symboltable.h" #include "stringtable.h" #ifndef O_BINARY #define O_BINARY 0 #endif #define Version "2004-12-29" #define Copyright   "Copyright (c) 2004 Greg Parkern"  "Copyright (c) 2001 David E. O'Brienn"  "Copyright (c) 1996-1998 John D. Polstran"  "All rights reserved.n"  "n"  "Redistribution and use in source and binary forms, with or withoutn"  "modification, are permitted provided that the following conditionsn"  "are met:n"  "1. Redistributions of source code must retain the above copyrightn"  "   notice, this list of conditions and the following disclaimer.n"  "2. Redistributions in binary form must reproduce the above copyrightn"  "   notice, this list of conditions and the following disclaimer in then"  "   documentation and/or other materials provided with the distribution.n"  "n"  "THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' ANDn"  "ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THEn"  "IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSEn" "ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLEn"  "FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIALn" "DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODSn"  "OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)n"  "HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICTn" "LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAYn"  "OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OFn"  "SUCH DAMAGE.n" int Verbose = 0; static void version(void) {     fprintf(stderr, "peal-postlink version %snn", Version);     fprintf(stderr, "%sn", Copyright); } static void usage(const char *name) {     fprintf(stderr,              "Usage: %s [-vV] [-l section] [-k funcname] [-K filename] [-t resType] [-s resID] [-o output] filenamen"             "   -V: print version infon"             "   -v: verbosen"
  2. "   -l <section>: create shared stubn"             "   -k <funcname>: only keep global symboln"
  3.             "   -K <filename>: only keep global symbols listed in file 'filename'n"             "   -t <resType>: set resource type for -s (default is 'armc')n"             "   -o <output>: write result to file 'output'n"             "   -s <resID>: write result in .ro format with one resource per ELF sectionn"             "            (default is .bin format with everything in one resource)n"             , name);     exit(1); } int main(int argc, char **argv) {     int fd;     int err;     struct stat sb;     int ch;     const char *infilename;     const char *outfilename = NULL;     const char *stubsection = NULL;
  4.     const char *selfname;     vector<string> *keepSymbols = NULL;     int printversion = 0;     int splitindex = -1;     const char *resType = "armc";     // Parse options      Verbose = 0;     selfname = strrchr(argv[0], '/');     if (!selfname) selfname = argv[0];     while ((ch = getopt(argc, argv, "vVk:K:o:s:t:l:")) != -1) {         switch (ch) {         case 'V':              printversion = 1;             break;         case 'v':             Verbose = 1;             break;         case 'l':
  5. stubsection = optarg;
  6. break;
  7. case 'k':
  8. if (!keepSymbols)
  9. keepSymbols = new vector<string>(0);
  10. keepSymbols->push_back(string(optarg));
  11. break;
  12.         case 'K': { FILE *keepfile = fopen(optarg, "r");
  13. if (!keepfile) { perror(optarg); return 1; }
  14. if (!keepSymbols)
  15. keepSymbols = new vector<string>(0);
  16. char buf[1024];
  17. char *keep;
  18. size_t len;
  19. while ((keep = fgets(buf, sizeof(buf), keepfile))) {
  20. char *end = strstr(keep, "n");
  21. if (end) len = end - keep;  // skip newline, if any
  22. else len = strlen(keep);
  23. keepSymbols->push_back(string(keep, len));
  24. }
  25.             break; }         case 'o':             if (outfilename) {                 fprintf(stderr, "%s: -o may be used only oncen", selfname);                 usage(selfname);             } else {                 outfilename = optarg;             }             break;         case 's':             if (splitindex != -1) {                 fprintf(stderr, "%s: -s may be used only oncen", selfname);                 usage(selfname);             } else {                 splitindex = atoi(optarg);             }             break;         case 't':             resType = optarg;             if (strlen(resType) != 4) {                 fprintf(stderr, "%s: -t resource type must be exactly four characters longn", selfname);                 usage(selfname);             }             break;         case '?':         default:             usage(selfname);             break;         }     }          argc -= optind;     argv += optind;     if (printversion) {         version();     }     if (argc == 0) {         fprintf(stderr, "%s: no file specifiedn", selfname);         usage(selfname);     }     infilename = argv[argc-1];     if (!outfilename) outfilename = infilename;     // Read file into memory     fd = open(infilename, O_RDONLY | O_BINARY, 0);     if (fd < 0) { perror(infilename); return 1; }     err = fstat(fd, &sb);     if (err) { perror(infilename); return 1; }     uint8_t *buf = (uint8_t *)malloc(sb.st_size);     if (sb.st_size != read(fd, buf, sb.st_size)) {          perror(infilename); return 1;     }     close(fd);     // Read the ELF image, undo relocations, and record new relocations.     Image image(buf);     // Scrub the symbol table.      // - keep all global symbols     // - keep a symbol for data and text sections     // - fixme apply export list here          image.symtab().strip(keepSymbols);     image.trimSections();  // removes all string tables
  26.     image.addSectionGlobals();
  27. image.buildStub(stubsection);     image.buildSymbolStringTable();     image.buildRelocations();
  28.     image.buildSectionStringTable();     image.write(resType, splitindex, outfilename);     // fixme paranoia checks:     // all ro+alloc+contents sections contiguous (in file and vm)     // no ro+alloc-contents sections     // all rw+alloc sections coniguous (in file and vm)     return 0; }