spelling fixes: arch/alpha/
[linux-2.6.git] / arch / ppc / boot / utils / mkbugboot.c
1 /*
2  * Makes a Motorola PPCBUG ROM bootable image which can be flashed
3  * into one of the FLASH banks on a Motorola PowerPlus board.
4  *
5  * Author: Matt Porter <mporter@mvista.com>
6  *
7  * 2001 (c) MontaVista, Software, Inc.  This file is licensed under
8  * the terms of the GNU General Public License version 2.  This program
9  * is licensed "as is" without any warranty of any kind, whether express
10  * or implied.
11  */
12
13 #define ELF_HEADER_SIZE 65536
14
15 #include <unistd.h>
16 #include <sys/stat.h>
17 #include <string.h>
18 #include <stdio.h>
19 #include <stdlib.h>
20 #include <errno.h>
21 #include <fcntl.h>
22 #include <netinet/in.h>
23 #ifdef __sun__
24 #include <inttypes.h>
25 #else
26 #include <stdint.h>
27 #endif
28
29 /* size of read buffer */
30 #define SIZE 0x1000
31
32 /* PPCBUG ROM boot header */
33 typedef struct bug_boot_header {
34   uint8_t       magic_word[4];          /* "BOOT" */
35   uint32_t      entry_offset;           /* Offset from top of header to code */
36   uint32_t      routine_length;         /* Length of code */
37   uint8_t       routine_name[8];        /* Name of the boot code */
38 } bug_boot_header_t;
39
40 #define HEADER_SIZE     sizeof(bug_boot_header_t)
41
42 void update_checksum(void *buf, size_t size, uint16_t *sum)
43 {
44         uint32_t csum = *sum;
45
46         while (size) {
47                 csum += *(uint16_t *)buf;
48                 if (csum > 0xffff)
49                         csum -= 0xffff;
50                 buf = (uint16_t *)buf + 1;
51                 size -= 2;
52         }
53         *sum = csum;
54 }
55
56 uint32_t copy_image(int in_fd, int out_fd, uint16_t *sum)
57 {
58         uint8_t buf[SIZE];
59         int offset = 0;
60         int n;
61         uint32_t image_size = 0;
62
63         lseek(in_fd, ELF_HEADER_SIZE, SEEK_SET);
64
65         /* Copy an image while recording its size */
66         while ( (n = read(in_fd, buf + offset, SIZE - offset)) > 0 ) {
67                 n += offset;
68                 offset = n & 1;
69                 n -= offset;
70                 image_size = image_size + n;
71                 /* who's going to deal with short writes? */
72                 write(out_fd, buf, n);
73                 update_checksum(buf, n, sum);
74                 if (offset)
75                         buf[0] = buf[n];
76         }
77
78         /* BUG romboot requires that our size is divisible by 2 */
79         /* align image to 2 byte boundary */
80         if (offset) {
81                 image_size += 2;
82                 buf[1] = '\0';
83                 write(out_fd, buf, 2);
84                 update_checksum(buf, 2, sum);
85         }
86         return image_size;
87 }
88
89 void write_bugboot_header(int out_fd, uint32_t boot_size, uint16_t *sum)
90 {
91         static bug_boot_header_t bbh = {
92                 .magic_word = "BOOT",
93                 .routine_name = "LINUXROM"
94         };
95
96         /* Fill in the PPCBUG ROM boot header */
97         bbh.entry_offset = htonl(HEADER_SIZE);  /* Entry address */
98         bbh.routine_length= htonl(HEADER_SIZE+boot_size+2);     /* Routine length */
99
100         /* Output the header and bootloader to the file */
101         write(out_fd, &bbh, sizeof(bug_boot_header_t));
102         update_checksum(&bbh, sizeof(bug_boot_header_t), sum);
103 }
104
105 int main(int argc, char *argv[])
106 {
107         int image_fd, bugboot_fd;
108         uint32_t kernel_size = 0;
109         uint16_t checksum = 0;
110
111         if (argc != 3) {
112                 fprintf(stderr, "usage: %s <kernel_image> <bugboot>\n",argv[0]);
113                 exit(-1);
114         }
115
116         /* Get file args */
117
118         /* kernel image file */
119         if ((image_fd = open(argv[1] , 0)) < 0)
120                 exit(-1);
121
122         /* bugboot file */
123         if (!strcmp(argv[2], "-"))
124                 bugboot_fd = 1;                 /* stdout */
125         else if ((bugboot_fd = creat(argv[2] , 0755)) < 0)
126                 exit(-1);
127
128         /* Set file position after ROM header block where zImage will be written */
129         lseek(bugboot_fd, HEADER_SIZE, SEEK_SET);
130
131         /* Copy kernel image into bugboot image */
132         kernel_size = copy_image(image_fd, bugboot_fd, &checksum);
133
134         /* Set file position to beginning where header/romboot will be written */
135         lseek(bugboot_fd, 0, SEEK_SET);
136
137         /* Write out BUG header/romboot */
138         write_bugboot_header(bugboot_fd, kernel_size, &checksum);
139
140         /* Write out the calculated checksum */
141         lseek(bugboot_fd, 0, SEEK_END);
142         write(bugboot_fd, &checksum, 2);
143
144         /* Close bugboot file */
145         close(bugboot_fd);
146         return 0;
147 }