Commit c32527a1 authored by Al Viro's avatar Al Viro Committed by Paul Mackerras

[POWERPC] rewrite mkprep and mkbugboot in sane C

Signed-off-by: default avatarAl Viro <viro@zeniv.linux.org.uk>
Signed-off-by: default avatarPaul Mackerras <paulus@samba.org>
parent 8c42ec2c
...@@ -19,36 +19,13 @@ ...@@ -19,36 +19,13 @@
#include <stdlib.h> #include <stdlib.h>
#include <errno.h> #include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <netinet/in.h>
#ifdef __sun__ #ifdef __sun__
#include <inttypes.h> #include <inttypes.h>
#else #else
#include <stdint.h> #include <stdint.h>
#endif #endif
#ifdef __i386__
#define cpu_to_be32(x) le32_to_cpu(x)
#define cpu_to_be16(x) le16_to_cpu(x)
#else
#define cpu_to_be32(x) (x)
#define cpu_to_be16(x) (x)
#endif
#define cpu_to_le32(x) le32_to_cpu((x))
unsigned long le32_to_cpu(unsigned long x)
{
return (((x & 0x000000ffU) << 24) |
((x & 0x0000ff00U) << 8) |
((x & 0x00ff0000U) >> 8) |
((x & 0xff000000U) >> 24));
}
#define cpu_to_le16(x) le16_to_cpu((x))
unsigned short le16_to_cpu(unsigned short x)
{
return (((x & 0x00ff) << 8) |
((x & 0xff00) >> 8));
}
/* size of read buffer */ /* size of read buffer */
#define SIZE 0x1000 #define SIZE 0x1000
...@@ -62,79 +39,76 @@ typedef struct bug_boot_header { ...@@ -62,79 +39,76 @@ typedef struct bug_boot_header {
#define HEADER_SIZE sizeof(bug_boot_header_t) #define HEADER_SIZE sizeof(bug_boot_header_t)
uint32_t copy_image(int32_t in_fd, int32_t out_fd) void update_checksum(void *buf, size_t size, uint16_t *sum)
{
uint32_t csum = *sum;
while (size) {
csum += *(uint16_t *)buf;
if (csum > 0xffff)
csum -= 0xffff;
buf = (uint16_t *)buf + 1;
size -= 2;
}
*sum = csum;
}
uint32_t copy_image(int in_fd, int out_fd, uint16_t *sum)
{ {
uint8_t buf[SIZE]; uint8_t buf[SIZE];
int offset = 0;
int n; int n;
uint32_t image_size = 0; uint32_t image_size = 0;
uint8_t zero = 0;
lseek(in_fd, ELF_HEADER_SIZE, SEEK_SET); lseek(in_fd, ELF_HEADER_SIZE, SEEK_SET);
/* Copy an image while recording its size */ /* Copy an image while recording its size */
while ( (n = read(in_fd, buf, SIZE)) > 0 ) while ( (n = read(in_fd, buf + offset, SIZE - offset)) > 0 ) {
{ n += offset;
offset = n & 1;
n -= offset;
image_size = image_size + n; image_size = image_size + n;
/* who's going to deal with short writes? */
write(out_fd, buf, n); write(out_fd, buf, n);
update_checksum(buf, n, sum);
if (offset)
buf[0] = buf[n];
} }
/* BUG romboot requires that our size is divisible by 2 */ /* BUG romboot requires that our size is divisible by 2 */
/* align image to 2 byte boundary */ /* align image to 2 byte boundary */
if (image_size % 2) if (offset) {
{ image_size += 2;
image_size++; buf[1] = '\0';
write(out_fd, &zero, 1); write(out_fd, buf, 2);
update_checksum(buf, 2, sum);
} }
return image_size; return image_size;
} }
void write_bugboot_header(int32_t out_fd, uint32_t boot_size) void write_bugboot_header(int out_fd, uint32_t boot_size, uint16_t *sum)
{ {
uint8_t header_block[HEADER_SIZE]; static bug_boot_header_t bbh = {
bug_boot_header_t *bbh = (bug_boot_header_t *)&header_block[0]; .magic_word = "BOOT",
.routine_name = "LINUXROM"
memset(header_block, 0, HEADER_SIZE); };
/* Fill in the PPCBUG ROM boot header */ /* Fill in the PPCBUG ROM boot header */
strncpy(bbh->magic_word, "BOOT", 4); /* PPCBUG magic word */ bbh.entry_offset = htonl(HEADER_SIZE); /* Entry address */
bbh->entry_offset = cpu_to_be32(HEADER_SIZE); /* Entry address */ bbh.routine_length= htonl(HEADER_SIZE+boot_size+2); /* Routine length */
bbh->routine_length= cpu_to_be32(HEADER_SIZE+boot_size+2); /* Routine length */
strncpy(bbh->routine_name, "LINUXROM", 8); /* Routine name */
/* Output the header and bootloader to the file */ /* Output the header and bootloader to the file */
write(out_fd, header_block, HEADER_SIZE); write(out_fd, &bbh, sizeof(bug_boot_header_t));
} update_checksum(&bbh, sizeof(bug_boot_header_t), sum);
uint16_t calc_checksum(int32_t bug_fd)
{
uint32_t checksum_var = 0;
uint8_t buf[2];
int n;
/* Checksum loop */
while ( (n = read(bug_fd, buf, 2) ) )
{
checksum_var = checksum_var + *(uint16_t *)buf;
/* If we carry out, mask it and add one to the checksum */
if (checksum_var >> 16)
checksum_var = (checksum_var & 0x0000ffff) + 1;
}
return checksum_var;
} }
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
int32_t image_fd, bugboot_fd; int image_fd, bugboot_fd;
int argptr = 1;
uint32_t kernel_size = 0; uint32_t kernel_size = 0;
uint16_t checksum = 0; uint16_t checksum = 0;
uint8_t bugbootname[256];
if ( (argc != 3) ) if (argc != 3) {
{
fprintf(stderr, "usage: %s <kernel_image> <bugboot>\n",argv[0]); fprintf(stderr, "usage: %s <kernel_image> <bugboot>\n",argv[0]);
exit(-1); exit(-1);
} }
...@@ -142,44 +116,32 @@ int main(int argc, char *argv[]) ...@@ -142,44 +116,32 @@ int main(int argc, char *argv[])
/* Get file args */ /* Get file args */
/* kernel image file */ /* kernel image file */
if ((image_fd = open( argv[argptr] , 0)) < 0) if ((image_fd = open(argv[1] , 0)) < 0)
exit(-1); exit(-1);
argptr++;
/* bugboot file */ /* bugboot file */
if ( !strcmp( argv[argptr], "-" ) ) if (!strcmp(argv[2], "-"))
bugboot_fd = 1; /* stdout */ bugboot_fd = 1; /* stdout */
else else if ((bugboot_fd = creat(argv[2] , 0755)) < 0)
if ((bugboot_fd = creat( argv[argptr] , 0755)) < 0)
exit(-1); exit(-1);
else
strcpy(bugbootname, argv[argptr]);
argptr++;
/* Set file position after ROM header block where zImage will be written */ /* Set file position after ROM header block where zImage will be written */
lseek(bugboot_fd, HEADER_SIZE, SEEK_SET); lseek(bugboot_fd, HEADER_SIZE, SEEK_SET);
/* Copy kernel image into bugboot image */ /* Copy kernel image into bugboot image */
kernel_size = copy_image(image_fd, bugboot_fd); kernel_size = copy_image(image_fd, bugboot_fd, &checksum);
close(image_fd);
/* Set file position to beginning where header/romboot will be written */ /* Set file position to beginning where header/romboot will be written */
lseek(bugboot_fd, 0, SEEK_SET); lseek(bugboot_fd, 0, SEEK_SET);
/* Write out BUG header/romboot */ /* Write out BUG header/romboot */
write_bugboot_header(bugboot_fd, kernel_size); write_bugboot_header(bugboot_fd, kernel_size, &checksum);
/* Close bugboot file */
close(bugboot_fd);
/* Reopen it as read/write */
bugboot_fd = open(bugbootname, O_RDWR);
/* Calculate checksum */
checksum = calc_checksum(bugboot_fd);
/* Write out the calculated checksum */ /* Write out the calculated checksum */
lseek(bugboot_fd, 0, SEEK_END);
write(bugboot_fd, &checksum, 2); write(bugboot_fd, &checksum, 2);
/* Close bugboot file */
close(bugboot_fd);
return 0; return 0;
} }
...@@ -15,161 +15,130 @@ ...@@ -15,161 +15,130 @@
* Modified for Sparc hosted builds by Peter Wahl <PeterWahl@web.de> * Modified for Sparc hosted builds by Peter Wahl <PeterWahl@web.de>
*/ */
#include <fcntl.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h>
#include <string.h> #include <string.h>
#include <strings.h> #include <stdlib.h>
#include <sys/stat.h>
#include <unistd.h>
#define cpu_to_le32(x) le32_to_cpu((x))
unsigned long le32_to_cpu(unsigned long x)
{
return (((x & 0x000000ffU) << 24) |
((x & 0x0000ff00U) << 8) |
((x & 0x00ff0000U) >> 8) |
((x & 0xff000000U) >> 24));
}
#define cpu_to_le16(x) le16_to_cpu((x))
unsigned short le16_to_cpu(unsigned short x)
{
return (((x & 0x00ff) << 8) |
((x & 0xff00) >> 8));
}
#define cpu_to_be32(x) (x)
#define be32_to_cpu(x) (x)
#define cpu_to_be16(x) (x)
#define be16_to_cpu(x) (x)
/* size of read buffer */ /* size of read buffer */
#define SIZE 0x1000 #define SIZE 0x1000
typedef unsigned long dword_t;
typedef unsigned short word_t;
typedef unsigned char byte_t;
typedef byte_t block_t[512];
typedef byte_t page_t[4096];
/* /*
* Partition table entry * Partition table entry
* - from the PReP spec * - from the PReP spec
*/ */
typedef struct partition_entry { typedef struct partition_entry {
byte_t boot_indicator; unsigned char boot_indicator;
byte_t starting_head; unsigned char starting_head;
byte_t starting_sector; unsigned char starting_sector;
byte_t starting_cylinder; unsigned char starting_cylinder;
byte_t system_indicator; unsigned char system_indicator;
byte_t ending_head; unsigned char ending_head;
byte_t ending_sector; unsigned char ending_sector;
byte_t ending_cylinder; unsigned char ending_cylinder;
dword_t beginning_sector; unsigned char beginning_sector[4];
dword_t number_of_sectors; unsigned char number_of_sectors[4];
} partition_entry_t; } partition_entry_t;
#define BootActive 0x80 #define BootActive 0x80
#define SystemPrep 0x41 #define SystemPrep 0x41
void copy_image(int , int); void copy_image(FILE *, FILE *);
void write_prep_partition(int , int ); void write_prep_partition(FILE *, FILE *);
void write_asm_data( int in, int out ); void write_asm_data(FILE *, FILE *);
unsigned int elfhdr_size = 65536; unsigned int elfhdr_size = 65536;
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
int in_fd, out_fd; FILE *in, *out;
int argptr = 1; int argptr = 1;
unsigned int prep = 0; int prep = 0;
unsigned int asmoutput = 0; int asmoutput = 0;
if ( (argc < 3) || (argc > 4) ) if (argc < 3 || argc > 4) {
{ fprintf(stderr, "usage: %s [-pbp] [-asm] <boot-file> <image>\n",
fprintf(stderr, "usage: %s [-pbp] [-asm] <boot-file> <image>\n",argv[0]); argv[0]);
exit(-1); exit(-1);
} }
/* needs to handle args more elegantly -- but this is a small/simple program */ /* needs to handle args more elegantly -- but this is a small/simple program */
/* check for -pbp */ /* check for -pbp */
if ( !strcmp( argv[argptr], "-pbp" ) ) if (!strcmp(argv[argptr], "-pbp")) {
{
prep = 1; prep = 1;
argptr++; argptr++;
} }
/* check for -asm */ /* check for -asm */
if ( !strcmp( argv[argptr], "-asm" ) ) if (!strcmp(argv[argptr], "-asm")) {
{
asmoutput = 1; asmoutput = 1;
argptr++; argptr++;
} }
/* input file */ /* input file */
if ( !strcmp( argv[argptr], "-" ) ) if (!strcmp(argv[argptr], "-"))
in_fd = 0; /* stdin */ in = stdin;
else else if (!(in = fopen(argv[argptr], "r")))
if ((in_fd = open( argv[argptr] , 0)) < 0)
exit(-1); exit(-1);
argptr++; argptr++;
/* output file */ /* output file */
if ( !strcmp( argv[argptr], "-" ) ) if (!strcmp(argv[argptr], "-"))
out_fd = 1; /* stdout */ out = stdout;
else else if (!(out = fopen(argv[argptr], "w")))
if ((out_fd = creat( argv[argptr] , 0755)) < 0)
exit(-1); exit(-1);
argptr++; argptr++;
/* skip elf header in input file */ /* skip elf header in input file */
/*if ( !prep )*/ /*if ( !prep )*/
lseek(in_fd, elfhdr_size, SEEK_SET); fseek(in, elfhdr_size, SEEK_SET);
/* write prep partition if necessary */ /* write prep partition if necessary */
if ( prep ) if (prep)
write_prep_partition( in_fd, out_fd ); write_prep_partition(in, out);
/* write input image to bootimage */ /* write input image to bootimage */
if ( asmoutput ) if (asmoutput)
write_asm_data( in_fd, out_fd ); write_asm_data(in, out);
else else
copy_image(in_fd, out_fd); copy_image(in, out);
return 0; return 0;
} }
void write_prep_partition(int in, int out) void store_le32(unsigned int v, unsigned char *p)
{
p[0] = v;
p[1] = v >>= 8;
p[2] = v >>= 8;
p[3] = v >> 8;
}
void write_prep_partition(FILE *in, FILE *out)
{ {
unsigned char block[512]; unsigned char block[512];
partition_entry_t pe; partition_entry_t pe;
dword_t *entry = (dword_t *)&block[0]; unsigned char *entry = block;
dword_t *length = (dword_t *)&block[sizeof(long)]; unsigned char *length = block + 4;
struct stat info; long pos = ftell(in), size;
if (fstat(in, &info) < 0) if (fseek(in, 0, SEEK_END) < 0) {
{ fprintf(stderr,"info failed\n");
exit(-1);
}
size = ftell(in);
if (fseek(in, pos, SEEK_SET) < 0) {
fprintf(stderr,"info failed\n"); fprintf(stderr,"info failed\n");
exit(-1); exit(-1);
} }
bzero( block, sizeof block ); memset(block, '\0', sizeof(block));
/* set entry point and boot image size skipping over elf header */ /* set entry point and boot image size skipping over elf header */
#ifdef __i386__ store_le32(0x400/*+65536*/, entry);
*entry = 0x400/*+65536*/; store_le32(size-elfhdr_size+0x400, length);
*length = info.st_size-elfhdr_size+0x400;
#else
*entry = cpu_to_le32(0x400/*+65536*/);
*length = cpu_to_le32(info.st_size-elfhdr_size+0x400);
#endif /* __i386__ */
/* sets magic number for msdos partition (used by linux) */ /* sets magic number for msdos partition (used by linux) */
block[510] = 0x55; block[510] = 0x55;
...@@ -203,91 +172,70 @@ void write_prep_partition(int in, int out) ...@@ -203,91 +172,70 @@ void write_prep_partition(int in, int out)
* - unlike the above sector numbers, the beginning sector is zero-based! * - unlike the above sector numbers, the beginning sector is zero-based!
*/ */
#if 0 #if 0
pe.beginning_sector = cpu_to_le32(1); store_le32(1, pe.beginning_sector);
#else #else
/* This has to be 0 on the PowerStack? */ /* This has to be 0 on the PowerStack? */
#ifdef __i386__ store_le32(0, pe.beginning_sector);
pe.beginning_sector = 0;
#else
pe.beginning_sector = cpu_to_le32(0);
#endif /* __i386__ */
#endif #endif
#ifdef __i386__ store_le32(2*18*80-1, pe.number_of_sectors);
pe.number_of_sectors = 2*18*80-1;
#else
pe.number_of_sectors = cpu_to_le32(2*18*80-1);
#endif /* __i386__ */
memcpy(&block[0x1BE], &pe, sizeof(pe)); memcpy(&block[0x1BE], &pe, sizeof(pe));
write( out, block, sizeof(block) ); fwrite(block, sizeof(block), 1, out);
write( out, entry, sizeof(*entry) ); fwrite(entry, 4, 1, out);
write( out, length, sizeof(*length) ); fwrite(length, 4, 1, out);
/* set file position to 2nd sector where image will be written */ /* set file position to 2nd sector where image will be written */
lseek( out, 0x400, SEEK_SET ); fseek( out, 0x400, SEEK_SET );
} }
void void copy_image(FILE *in, FILE *out)
copy_image(int in, int out)
{ {
char buf[SIZE]; char buf[SIZE];
int n; int n;
while ( (n = read(in, buf, SIZE)) > 0 ) while ( (n = fread(buf, 1, SIZE, in)) > 0 )
write(out, buf, n); fwrite(buf, 1, n, out);
} }
void void
write_asm_data( int in, int out ) write_asm_data(FILE *in, FILE *out)
{ {
int i, cnt, pos, len; int i, cnt, pos = 0;
unsigned int cksum, val; unsigned int cksum = 0, val;
unsigned char *lp; unsigned char *lp;
unsigned char buf[SIZE]; unsigned char buf[SIZE];
unsigned char str[256]; size_t len;
write( out, "\t.data\n\t.globl input_data\ninput_data:\n", fputs("\t.data\n\t.globl input_data\ninput_data:\n", out);
strlen( "\t.data\n\t.globl input_data\ninput_data:\n" ) ); while ((len = fread(buf, 1, sizeof(buf), in)) > 0) {
pos = 0;
cksum = 0;
while ((len = read(in, buf, sizeof(buf))) > 0)
{
cnt = 0; cnt = 0;
lp = (unsigned char *)buf; lp = buf;
len = (len + 3) & ~3; /* Round up to longwords */ /* Round up to longwords */
for (i = 0; i < len; i += 4) while (len & 3)
{ buf[len++] = '\0';
for (i = 0; i < len; i += 4) {
if (cnt == 0) if (cnt == 0)
{ fputs("\t.long\t", out);
write( out, "\t.long\t", strlen( "\t.long\t" ) ); fprintf(out, "0x%02X%02X%02X%02X",
} lp[0], lp[1], lp[2], lp[3]);
sprintf( str, "0x%02X%02X%02X%02X", lp[0], lp[1], lp[2], lp[3]);
write( out, str, strlen(str) );
val = *(unsigned long *)lp; val = *(unsigned long *)lp;
cksum ^= val; cksum ^= val;
lp += 4; lp += 4;
if (++cnt == 4) if (++cnt == 4) {
{
cnt = 0; cnt = 0;
sprintf( str, " # %x \n", pos+i-12); fprintf(out, " # %x \n", pos+i-12);
write( out, str, strlen(str) ); } else {
} else fputs(",", out);
{
write( out, ",", 1 );
} }
} }
if (cnt) if (cnt)
{ fputs("0\n", out);
write( out, "0\n", 2 );
}
pos += len; pos += len;
} }
sprintf(str, "\t.globl input_len\ninput_len:\t.long\t0x%x\n", pos); fprintf(out, "\t.globl input_len\ninput_len:\t.long\t0x%x\n", pos);
write( out, str, strlen(str) );
fprintf(stderr, "cksum = %x\n", cksum); fprintf(stderr, "cksum = %x\n", cksum);
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment