objstrip.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:6k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * arch/alpha/boot/tools/objstrip.c
  3.  *
  4.  * Strip the object file headers/trailers from an executable (ELF or ECOFF).
  5.  *
  6.  * Copyright (C) 1996 David Mosberger-Tang.
  7.  */
  8. /*
  9.  * Converts an ECOFF or ELF object file into a bootable file.  The
  10.  * object file must be a OMAGIC file (i.e., data and bss follow immediatly
  11.  * behind the text).  See DEC "Assembly Language Programmer's Guide"
  12.  * documentation for details.  The SRM boot process is documented in
  13.  * the Alpha AXP Architecture Reference Manual, Second Edition by
  14.  * Richard L. Sites and Richard T. Witek.
  15.  */
  16. #include <stdio.h>
  17. #include <string.h>
  18. #include <stdlib.h>
  19. #include <unistd.h>
  20. #include <sys/fcntl.h>
  21. #include <sys/stat.h>
  22. #include <sys/types.h>
  23. #include <linux/a.out.h>
  24. #include <linux/coff.h>
  25. #include <linux/param.h>
  26. #include <linux/string.h>
  27. #ifdef __ELF__
  28. # include <asm/elf.h>
  29. # include <linux/elf.h>
  30. #endif
  31. /* bootfile size must be multiple of BLOCK_SIZE: */
  32. #define BLOCK_SIZE 512
  33. const char * prog_name;
  34. void
  35. usage (void)
  36. {
  37.     fprintf(stderr,
  38.     "usage: %s [-v] -p file primaryn"
  39.     "       %s [-vb] file [secondary]n", prog_name, prog_name);
  40.     exit(1);
  41. }
  42. int
  43. main (int argc, char *argv[])
  44. {
  45.     size_t nwritten, tocopy, n, mem_size, fil_size, pad = 0;
  46.     int fd, ofd, i, j, verbose = 0, primary = 0;
  47.     char buf[8192], *inname;
  48.     struct exec * aout; /* includes file & aout header */
  49.     long offset;
  50. #ifdef __ELF__
  51.     struct elfhdr *elf;
  52.     struct elf_phdr *elf_phdr; /* program header */
  53.     unsigned long long e_entry;
  54. #endif
  55.     prog_name = argv[0];
  56.     for (i = 1; i < argc && argv[i][0] == '-'; ++i) {
  57. for (j = 1; argv[i][j]; ++j) {
  58.     switch (argv[i][j]) {
  59.       case 'v':
  60.   verbose = ~verbose;
  61.   break;
  62.       case 'b':
  63.   pad = BLOCK_SIZE;
  64.   break;
  65.       case 'p':
  66.   primary = 1; /* make primary bootblock */
  67.   break;
  68.     }
  69. }
  70.     }
  71.     if (i >= argc) {
  72. usage();
  73.     }
  74.     inname = argv[i++];
  75.     fd = open(inname, O_RDONLY);
  76.     if (fd == -1) {
  77. perror("open");
  78. exit(1);
  79.     }
  80.     ofd = 1;
  81.     if (i < argc) {
  82. ofd = open(argv[i++], O_WRONLY | O_CREAT | O_TRUNC, 0666);
  83. if (fd == -1) {
  84.     perror("open");
  85.     exit(1);
  86. }
  87.     }
  88.     if (primary) {
  89. /* generate bootblock for primary loader */
  90. unsigned long bb[64], sum = 0;
  91. struct stat st;
  92. off_t size;
  93. int i;
  94. if (ofd == 1) {
  95.     usage();
  96. }
  97. if (fstat(fd, &st) == -1) {
  98.     perror("fstat");
  99.     exit(1);
  100. }
  101. size = (st.st_size + BLOCK_SIZE - 1) & ~(BLOCK_SIZE - 1);
  102. memset(bb, 0, sizeof(bb));
  103. strcpy((char *) bb, "Linux SRM bootblock");
  104. bb[60] = size / BLOCK_SIZE; /* count */
  105. bb[61] = 1; /* starting sector # */
  106. bb[62] = 0; /* flags---must be 0 */
  107. for (i = 0; i < 63; ++i) {
  108.     sum += bb[i];
  109. }
  110. bb[63] = sum;
  111. if (write(ofd, bb, sizeof(bb)) != sizeof(bb)) {
  112.     perror("boot-block write");
  113.     exit(1);
  114. }
  115. printf("%lun", size);
  116. return 0;
  117.     }
  118.     /* read and inspect exec header: */
  119.     if (read(fd, buf, sizeof(buf)) < 0) {
  120. perror("read");
  121. exit(1);
  122.     }
  123. #ifdef __ELF__
  124.     elf = (struct elfhdr *) buf;
  125.     if (elf->e_ident[0] == 0x7f && strncmp(elf->e_ident + 1, "ELF", 3) == 0) {
  126. if (elf->e_type != ET_EXEC) {
  127.     fprintf(stderr, "%s: %s is not an ELF executablen",
  128.     prog_name, inname);
  129.     exit(1);
  130. }
  131. if (!elf_check_arch(elf)) {
  132.     fprintf(stderr, "%s: is not for this processor (e_machine=%d)n",
  133.     prog_name, elf->e_machine);
  134.     exit(1);
  135. }
  136. if (elf->e_phnum != 1) {
  137.     fprintf(stderr,
  138.     "%s: %d program headers (forgot to link with -N?)n",
  139.     prog_name, elf->e_phnum);
  140. }
  141. e_entry = elf->e_entry;
  142. lseek(fd, elf->e_phoff, SEEK_SET);
  143. if (read(fd, buf, sizeof(*elf_phdr)) != sizeof(*elf_phdr)) {
  144.     perror("read");
  145.     exit(1);
  146. }
  147. elf_phdr = (struct elf_phdr *) buf;
  148. offset  = elf_phdr->p_offset;
  149. mem_size = elf_phdr->p_memsz;
  150. fil_size = elf_phdr->p_filesz;
  151. /* work around ELF bug: */
  152. if (elf_phdr->p_vaddr < e_entry) {
  153.     unsigned long delta = e_entry - elf_phdr->p_vaddr;
  154.     offset   += delta;
  155.     mem_size -= delta;
  156.     fil_size -= delta;
  157.     elf_phdr->p_vaddr += delta;
  158. }
  159. if (verbose) {
  160.     fprintf(stderr, "%s: extracting %#016lx-%#016lx (at %lx)n",
  161.     prog_name, (long) elf_phdr->p_vaddr,
  162.     elf_phdr->p_vaddr + fil_size, offset);
  163. }
  164.     } else
  165. #endif
  166.     {
  167. aout = (struct exec *) buf;
  168. if (!(aout->fh.f_flags & COFF_F_EXEC)) {
  169.     fprintf(stderr, "%s: %s is not in executable formatn",
  170.     prog_name, inname);
  171.     exit(1);
  172. }
  173. if (aout->fh.f_opthdr != sizeof(aout->ah)) {
  174.     fprintf(stderr, "%s: %s has unexpected optional header sizen",
  175.     prog_name, inname);
  176.     exit(1);
  177. }
  178. if (N_MAGIC(*aout) != OMAGIC) {
  179.     fprintf(stderr, "%s: %s is not an OMAGIC filen",
  180.     prog_name, inname);
  181.     exit(1);
  182. }
  183. offset = N_TXTOFF(*aout);
  184. fil_size = aout->ah.tsize + aout->ah.dsize;
  185. mem_size = fil_size + aout->ah.bsize;
  186. if (verbose) {
  187.     fprintf(stderr, "%s: extracting %#016lx-%#016lx (at %lx)n",
  188.     prog_name, aout->ah.text_start,
  189.     aout->ah.text_start + fil_size, offset);
  190. }
  191.     }
  192.     if (lseek(fd, offset, SEEK_SET) != offset) {
  193. perror("lseek");
  194. exit(1);
  195.     }
  196.     if (verbose) {
  197. fprintf(stderr, "%s: copying %lu byte from %sn",
  198. prog_name, (unsigned long) fil_size, inname);
  199.     }
  200.     tocopy = fil_size;
  201.     while (tocopy > 0) {
  202. n = tocopy;
  203. if (n > sizeof(buf)) {
  204.     n = sizeof(buf);
  205. }
  206. tocopy -= n;
  207. if ((size_t) read(fd, buf, n) != n) {
  208.     perror("read");
  209.     exit(1);
  210. }
  211. do {
  212.     nwritten = write(ofd, buf, n);
  213.     if ((ssize_t) nwritten == -1) {
  214. perror("write");
  215. exit(1);
  216.     }
  217.     n -= nwritten;
  218. } while (n > 0);
  219.     }
  220.     if (pad) {
  221. mem_size = ((mem_size + pad - 1) / pad) * pad;
  222.     }
  223.     tocopy = mem_size - fil_size;
  224.     if (tocopy > 0) {
  225. fprintf(stderr,
  226. "%s: zero-filling bss and aligning to %lu with %lu bytesn",
  227. prog_name, pad, (unsigned long) tocopy);
  228. memset(buf, 0x00, sizeof(buf));
  229. do {
  230.     n = tocopy;
  231.     if (n > sizeof(buf)) {
  232. n = sizeof(buf);
  233.     }
  234.     nwritten = write(ofd, buf, n);
  235.     if ((ssize_t) nwritten == -1) {
  236. perror("write");
  237. exit(1);
  238.     }
  239.     tocopy -= nwritten;
  240. } while (tocopy > 0);
  241.     }
  242.     return 0;
  243. }