PageRenderTime 15ms CodeModel.GetById 5ms app.highlight 6ms RepoModel.GetById 2ms app.codeStats 0ms

/arch/mips/boot/addinitrd.c

https://bitbucket.org/evzijst/gittest
C | 131 lines | 101 code | 13 blank | 17 comment | 41 complexity | fc8099e10cf2bf4f4ec18de0fd263a81 MD5 | raw file
  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
 35void die (char *s)
 36{
 37	perror (s);
 38	exit (1);
 39}
 40
 41int 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}