arch/mips/boot/addinitrd.c

   1 /*
   2  * addinitrd - program to add a initrd image to an ecoff kernel
   3  *
   4  * (C) 1999 Thomas Bogendoerfer
   5  * minor modifications, cleanup: Guido Guenther <agx@sigxcpu.org>
   6  * further cleanup: Maciej W. Rozycki
   7  */
   8
   9 #include <sys/types.h>
  10 #include <sys/stat.h>
  11 #include <fcntl.h>
  12 #include <unistd.h>
  13 #include <stdio.h>
  14 #include <netinet/in.h>
  15
  16 #include "ecoff.h"
  17
  18 #define MIPS_PAGE_SIZE  4096
  19 #define MIPS_PAGE_MASK  (MIPS_PAGE_SIZE-1)
  20
  21 #define swab16(x) \
  22         ((unsigned short)( \
  23                 (((unsigned short)(x) & (unsigned short)0x00ffU) << 8) | \
  24                 (((unsigned short)(x) & (unsigned short)0xff00U) >> 8) ))
  25
  26 #define swab32(x) \
  27         ((unsigned int)( \
  28                 (((unsigned int)(x) & (unsigned int)0x000000ffUL) << 24) | \
  29                 (((unsigned int)(x) & (unsigned int)0x0000ff00UL) <<  8) | \
  30                 (((unsigned int)(x) & (unsigned int)0x00ff0000UL) >>  8) | \
  31                 (((unsigned int)(x) & (unsigned int)0xff000000UL) >> 24) ))
  32
  33 #define SWAB(a) (swab ? swab32(a) : (a))
  34
  35 void die(char *s)
  36 {
  37         perror(s);
  38         exit(1);
  39 }
  40
  41 int main(int argc, char *argv[])
  42 {
  43         int fd_vmlinux, fd_initrd, fd_outfile;
  44         FILHDR efile;
  45         AOUTHDR eaout;
  46         SCNHDR esecs[3];
  47         struct stat st;
  48         char buf[1024];
  49         unsigned long loadaddr;
  50         unsigned long initrd_header[2];
  51         int i, cnt;
  52         int swab = 0;
  53
  54         if (argc != 4) {
  55                 printf("Usage: %s <vmlinux> <initrd> <outfile>\n", argv[0]);
  56                 exit(1);
  57         }
  58
  59         if ((fd_vmlinux = open (argv[1], O_RDONLY)) < 0)
  60                  die("open vmlinux");
  61         if (read (fd_vmlinux, &efile, sizeof efile) != sizeof efile)
  62                 die("read file header");
  63         if (read (fd_vmlinux, &eaout, sizeof eaout) != sizeof eaout)
  64                 die("read aout header");
  65         if (read (fd_vmlinux, esecs, sizeof esecs) != sizeof esecs)
  66                 die("read section headers");
  67         /*
  68          * check whether the file is good for us
  69          */
  70         /* TBD */
  71
  72         /*
  73          * check, if we have to swab words
  74          */
  75         if (ntohs(0xaa55) == 0xaa55) {
  76                 if (efile.f_magic == swab16(MIPSELMAGIC))
  77                         swab = 1;
  78         } else {
  79                 if (efile.f_magic == swab16(MIPSEBMAGIC))
  80                         swab = 1;
  81         }
  82
  83         /* make sure we have an empty data segment for the initrd */
  84         if (eaout.dsize || esecs[1].s_size) {
  85                 fprintf(stderr, "Data segment not empty. Giving up!\n");
  86                 exit(1);
  87         }
  88         if ((fd_initrd = open (argv[2], O_RDONLY)) < 0)
  89                 die("open initrd");
  90         if (fstat (fd_initrd, &st) < 0)
  91                 die("fstat initrd");
  92         loadaddr = ((SWAB(esecs[2].s_vaddr) + SWAB(esecs[2].s_size)
  93                         + MIPS_PAGE_SIZE-1) & ~MIPS_PAGE_MASK) - 8;
  94         if (loadaddr < (SWAB(esecs[2].s_vaddr) + SWAB(esecs[2].s_size)))
  95                 loadaddr += MIPS_PAGE_SIZE;
  96         initrd_header[0] = SWAB(0x494E5244);
  97         initrd_header[1] = SWAB(st.st_size);
  98         eaout.dsize = esecs[1].s_size = initrd_header[1] = SWAB(st.st_size+8);
  99         eaout.data_start = esecs[1].s_vaddr = esecs[1].s_paddr = SWAB(loadaddr);
 100
 101         if ((fd_outfile = open (argv[3], O_RDWR|O_CREAT|O_TRUNC, 0666)) < 0)
 102                 die("open outfile");
 103         if (write (fd_outfile, &efile, sizeof efile) != sizeof efile)
 104                 die("write file header");
 105         if (write (fd_outfile, &eaout, sizeof eaout) != sizeof eaout)
 106                 die("write aout header");
 107         if (write (fd_outfile, esecs, sizeof esecs) != sizeof esecs)
 108                 die("write section headers");
 109         /* skip padding */
 110         if(lseek(fd_vmlinux, SWAB(esecs[0].s_scnptr), SEEK_SET) == (off_t)-1)
 111                 die("lseek vmlinux");
 112         if(lseek(fd_outfile, SWAB(esecs[0].s_scnptr), SEEK_SET) == (off_t)-1)
 113                 die("lseek outfile");
 114         /* copy text segment */
 115         cnt = SWAB(eaout.tsize);
 116         while (cnt) {
 117                 if ((i = read (fd_vmlinux, buf, sizeof buf)) <= 0)
 118                         die("read vmlinux");
 119                 if (write (fd_outfile, buf, i) != i)
 120                         die("write vmlinux");
 121                 cnt -= i;
 122         }
 123         if (write (fd_outfile, initrd_header, sizeof initrd_header) != sizeof initrd_header)
 124                 die("write initrd header");
 125         while ((i = read (fd_initrd, buf, sizeof buf)) > 0)
 126                 if (write (fd_outfile, buf, i) != i)
 127                         die("write initrd");
 128         close(fd_vmlinux);
 129         close(fd_initrd);
 130         return 0;
 131 }