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 
die(char * s)35 void die (char *s)
36 {
37 	perror (s);
38 	exit (1);
39 }
40 
main(int argc,char * argv[])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 }
132