提交 c32527a1 编写于 作者: A Al Viro 提交者: Paul Mackerras

[POWERPC] rewrite mkprep and mkbugboot in sane C

Signed-off-by: NAl Viro <viro@zeniv.linux.org.uk>
Signed-off-by: NPaul Mackerras <paulus@samba.org>
上级 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,124 +39,109 @@ typedef struct bug_boot_header { ...@@ -62,124 +39,109 @@ 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)
{ {
uint8_t buf[SIZE]; uint32_t csum = *sum;
int n;
uint32_t image_size = 0; while (size) {
uint8_t zero = 0; csum += *(uint16_t *)buf;
if (csum > 0xffff)
lseek(in_fd, ELF_HEADER_SIZE, SEEK_SET); csum -= 0xffff;
buf = (uint16_t *)buf + 1;
/* Copy an image while recording its size */ size -= 2;
while ( (n = read(in_fd, buf, SIZE)) > 0 ) }
{ *sum = csum;
image_size = image_size + n;
write(out_fd, buf, n);
}
/* BUG romboot requires that our size is divisible by 2 */
/* align image to 2 byte boundary */
if (image_size % 2)
{
image_size++;
write(out_fd, &zero, 1);
}
return image_size;
} }
void write_bugboot_header(int32_t out_fd, uint32_t boot_size) uint32_t copy_image(int in_fd, int out_fd, uint16_t *sum)
{ {
uint8_t header_block[HEADER_SIZE]; uint8_t buf[SIZE];
bug_boot_header_t *bbh = (bug_boot_header_t *)&header_block[0]; int offset = 0;
int n;
memset(header_block, 0, HEADER_SIZE); uint32_t image_size = 0;
/* Fill in the PPCBUG ROM boot header */ lseek(in_fd, ELF_HEADER_SIZE, SEEK_SET);
strncpy(bbh->magic_word, "BOOT", 4); /* PPCBUG magic word */
bbh->entry_offset = cpu_to_be32(HEADER_SIZE); /* Entry address */ /* Copy an image while recording its size */
bbh->routine_length= cpu_to_be32(HEADER_SIZE+boot_size+2); /* Routine length */ while ( (n = read(in_fd, buf + offset, SIZE - offset)) > 0 ) {
strncpy(bbh->routine_name, "LINUXROM", 8); /* Routine name */ n += offset;
offset = n & 1;
/* Output the header and bootloader to the file */ n -= offset;
write(out_fd, header_block, HEADER_SIZE); image_size = image_size + n;
/* who's going to deal with short writes? */
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 */
/* align image to 2 byte boundary */
if (offset) {
image_size += 2;
buf[1] = '\0';
write(out_fd, buf, 2);
update_checksum(buf, 2, sum);
}
return image_size;
} }
uint16_t calc_checksum(int32_t bug_fd) void write_bugboot_header(int out_fd, uint32_t boot_size, uint16_t *sum)
{ {
uint32_t checksum_var = 0; static bug_boot_header_t bbh = {
uint8_t buf[2]; .magic_word = "BOOT",
int n; .routine_name = "LINUXROM"
};
/* Checksum loop */
while ( (n = read(bug_fd, buf, 2) ) ) /* Fill in the PPCBUG ROM boot header */
{ bbh.entry_offset = htonl(HEADER_SIZE); /* Entry address */
checksum_var = checksum_var + *(uint16_t *)buf; bbh.routine_length= htonl(HEADER_SIZE+boot_size+2); /* Routine length */
/* If we carry out, mask it and add one to the checksum */ /* Output the header and bootloader to the file */
if (checksum_var >> 16) write(out_fd, &bbh, sizeof(bug_boot_header_t));
checksum_var = (checksum_var & 0x0000ffff) + 1; update_checksum(&bbh, sizeof(bug_boot_header_t), sum);
}
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) )
{
fprintf(stderr, "usage: %s <kernel_image> <bugboot>\n",argv[0]);
exit(-1);
}
/* Get file args */
/* kernel image file */
if ((image_fd = open( argv[argptr] , 0)) < 0)
exit(-1);
argptr++;
/* bugboot file */ if (argc != 3) {
if ( !strcmp( argv[argptr], "-" ) ) fprintf(stderr, "usage: %s <kernel_image> <bugboot>\n",argv[0]);
bugboot_fd = 1; /* stdout */ exit(-1);
else }
if ((bugboot_fd = creat( argv[argptr] , 0755)) < 0)
exit(-1);
else
strcpy(bugbootname, argv[argptr]);
argptr++;
/* Set file position after ROM header block where zImage will be written */ /* Get file args */
lseek(bugboot_fd, HEADER_SIZE, SEEK_SET);
/* Copy kernel image into bugboot image */ /* kernel image file */
kernel_size = copy_image(image_fd, bugboot_fd); if ((image_fd = open(argv[1] , 0)) < 0)
close(image_fd); exit(-1);
/* Set file position to beginning where header/romboot will be written */ /* bugboot file */
lseek(bugboot_fd, 0, SEEK_SET); if (!strcmp(argv[2], "-"))
bugboot_fd = 1; /* stdout */
else if ((bugboot_fd = creat(argv[2] , 0755)) < 0)
exit(-1);
/* Write out BUG header/romboot */ /* Set file position after ROM header block where zImage will be written */
write_bugboot_header(bugboot_fd, kernel_size); lseek(bugboot_fd, HEADER_SIZE, SEEK_SET);
/* Close bugboot file */ /* Copy kernel image into bugboot image */
close(bugboot_fd); kernel_size = copy_image(image_fd, bugboot_fd, &checksum);
/* Reopen it as read/write */ /* Set file position to beginning where header/romboot will be written */
bugboot_fd = open(bugbootname, O_RDWR); lseek(bugboot_fd, 0, SEEK_SET);
/* Calculate checksum */ /* Write out BUG header/romboot */
checksum = calc_checksum(bugboot_fd); write_bugboot_header(bugboot_fd, kernel_size, &checksum);
/* Write out the calculated checksum */ /* Write out the calculated checksum */
write(bugboot_fd, &checksum, 2); lseek(bugboot_fd, 0, SEEK_END);
write(bugboot_fd, &checksum, 2);
return 0; /* Close bugboot file */
close(bugboot_fd);
return 0;
} }
...@@ -15,279 +15,227 @@ ...@@ -15,279 +15,227 @@
* 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;
{ argptr++;
asmoutput = 1; }
argptr++;
} /* input file */
if (!strcmp(argv[argptr], "-"))
/* input file */ in = stdin;
if ( !strcmp( argv[argptr], "-" ) ) else if (!(in = fopen(argv[argptr], "r")))
in_fd = 0; /* stdin */ exit(-1);
else argptr++;
if ((in_fd = open( argv[argptr] , 0)) < 0)
exit(-1); /* output file */
argptr++; if (!strcmp(argv[argptr], "-"))
out = stdout;
/* output file */ else if (!(out = fopen(argv[argptr], "w")))
if ( !strcmp( argv[argptr], "-" ) ) exit(-1);
out_fd = 1; /* stdout */ argptr++;
else
if ((out_fd = creat( argv[argptr] , 0755)) < 0) /* skip elf header in input file */
exit(-1); /*if ( !prep )*/
argptr++; fseek(in, elfhdr_size, SEEK_SET);
/* skip elf header in input file */ /* write prep partition if necessary */
/*if ( !prep )*/ if (prep)
lseek(in_fd, elfhdr_size, SEEK_SET); write_prep_partition(in, out);
/* write prep partition if necessary */ /* write input image to bootimage */
if ( prep ) if (asmoutput)
write_prep_partition( in_fd, out_fd ); write_asm_data(in, out);
else
/* write input image to bootimage */ copy_image(in, out);
if ( asmoutput )
write_asm_data( in_fd, out_fd ); return 0;
else
copy_image(in_fd, out_fd);
return 0;
} }
void write_prep_partition(int in, int out) void store_le32(unsigned int v, unsigned char *p)
{ {
unsigned char block[512]; p[0] = v;
partition_entry_t pe; p[1] = v >>= 8;
dword_t *entry = (dword_t *)&block[0]; p[2] = v >>= 8;
dword_t *length = (dword_t *)&block[sizeof(long)]; p[3] = v >> 8;
struct stat info; }
if (fstat(in, &info) < 0) void write_prep_partition(FILE *in, FILE *out)
{ {
fprintf(stderr,"info failed\n"); unsigned char block[512];
exit(-1); partition_entry_t pe;
} unsigned char *entry = block;
unsigned char *length = block + 4;
bzero( block, sizeof block ); long pos = ftell(in), size;
/* set entry point and boot image size skipping over elf header */ if (fseek(in, 0, SEEK_END) < 0) {
#ifdef __i386__ fprintf(stderr,"info failed\n");
*entry = 0x400/*+65536*/; exit(-1);
*length = info.st_size-elfhdr_size+0x400; }
#else size = ftell(in);
*entry = cpu_to_le32(0x400/*+65536*/); if (fseek(in, pos, SEEK_SET) < 0) {
*length = cpu_to_le32(info.st_size-elfhdr_size+0x400); fprintf(stderr,"info failed\n");
#endif /* __i386__ */ exit(-1);
}
/* sets magic number for msdos partition (used by linux) */
block[510] = 0x55; memset(block, '\0', sizeof(block));
block[511] = 0xAA;
/* set entry point and boot image size skipping over elf header */
/* store_le32(0x400/*+65536*/, entry);
* Build a "PReP" partition table entry in the boot record store_le32(size-elfhdr_size+0x400, length);
* - "PReP" may only look at the system_indicator
*/ /* sets magic number for msdos partition (used by linux) */
pe.boot_indicator = BootActive; block[510] = 0x55;
pe.system_indicator = SystemPrep; block[511] = 0xAA;
/*
* The first block of the diskette is used by this "boot record" which /*
* actually contains the partition table. (The first block of the * Build a "PReP" partition table entry in the boot record
* partition contains the boot image, but I digress...) We'll set up * - "PReP" may only look at the system_indicator
* one partition on the diskette and it shall contain the rest of the */
* diskette. pe.boot_indicator = BootActive;
*/ pe.system_indicator = SystemPrep;
pe.starting_head = 0; /* zero-based */ /*
pe.starting_sector = 2; /* one-based */ * The first block of the diskette is used by this "boot record" which
pe.starting_cylinder = 0; /* zero-based */ * actually contains the partition table. (The first block of the
pe.ending_head = 1; /* assumes two heads */ * partition contains the boot image, but I digress...) We'll set up
pe.ending_sector = 18; /* assumes 18 sectors/track */ * one partition on the diskette and it shall contain the rest of the
pe.ending_cylinder = 79; /* assumes 80 cylinders/diskette */ * diskette.
*/
/* pe.starting_head = 0; /* zero-based */
* The "PReP" software ignores the above fields and just looks at pe.starting_sector = 2; /* one-based */
* the next two. pe.starting_cylinder = 0; /* zero-based */
* - size of the diskette is (assumed to be) pe.ending_head = 1; /* assumes two heads */
* (2 tracks/cylinder)(18 sectors/tracks)(80 cylinders/diskette) pe.ending_sector = 18; /* assumes 18 sectors/track */
* - unlike the above sector numbers, the beginning sector is zero-based! pe.ending_cylinder = 79; /* assumes 80 cylinders/diskette */
*/
/*
* The "PReP" software ignores the above fields and just looks at
* the next two.
* - size of the diskette is (assumed to be)
* (2 tracks/cylinder)(18 sectors/tracks)(80 cylinders/diskette)
* - 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
/* This has to be 0 on the PowerStack? */
#ifdef __i386__
pe.beginning_sector = 0;
#else #else
pe.beginning_sector = cpu_to_le32(0); /* This has to be 0 on the PowerStack? */
#endif /* __i386__ */ store_le32(0, pe.beginning_sector);
#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; cnt = 0;
cksum = 0; lp = buf;
while ((len = read(in, buf, sizeof(buf))) > 0) /* Round up to longwords */
{ while (len & 3)
cnt = 0; buf[len++] = '\0';
lp = (unsigned char *)buf; for (i = 0; i < len; i += 4) {
len = (len + 3) & ~3; /* Round up to longwords */ if (cnt == 0)
for (i = 0; i < len; i += 4) fputs("\t.long\t", out);
{ fprintf(out, "0x%02X%02X%02X%02X",
if (cnt == 0) lp[0], lp[1], lp[2], lp[3]);
{ val = *(unsigned long *)lp;
write( out, "\t.long\t", strlen( "\t.long\t" ) ); cksum ^= val;
} lp += 4;
sprintf( str, "0x%02X%02X%02X%02X", lp[0], lp[1], lp[2], lp[3]); if (++cnt == 4) {
write( out, str, strlen(str) ); cnt = 0;
val = *(unsigned long *)lp; fprintf(out, " # %x \n", pos+i-12);
cksum ^= val; } else {
lp += 4; fputs(",", out);
if (++cnt == 4) }
{ }
cnt = 0; if (cnt)
sprintf( str, " # %x \n", pos+i-12); fputs("0\n", out);
write( out, str, strlen(str) ); pos += len;
} else }
{ fprintf(out, "\t.globl input_len\ninput_len:\t.long\t0x%x\n", pos);
write( out, ",", 1 ); fprintf(stderr, "cksum = %x\n", cksum);
}
}
if (cnt)
{
write( out, "0\n", 2 );
}
pos += len;
}
sprintf(str, "\t.globl input_len\ninput_len:\t.long\t0x%x\n", pos);
write( out, str, strlen(str) );
fprintf(stderr, "cksum = %x\n", cksum);
} }
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册