devs/foxbox
1
0
Fork 0
forked from len0rd/rockbox
Code Pull requests Activity

Start to implement some functionality. You can now read/write the entire partition (as ipodpatcher used to do), extract the original firmware to an "apple_os.ipod" file, or replace the original firmware with either a Rockbox bootloader or Rockbox itself. Adding/Removing a bootloader isn't yet implemented.

Browse source 
git-svn-id: svn://svn.rockbox.org/rockbox/trunk@11755 a1c6a512-1295-4272-9138-f99709370657
This commit is contained in:
Dave Chapman 2006-12-14 01:24:05 +00:00
parent c657504e58
commit c5e30e15ec
1 changed files with 418 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

484
tools/ipodpatcher/ipodpatcher.c
View file

@ -59,8 +59,9 @@ static const char *apple_stop_sign = "{{~~ /-----\\ "\
and initialise it with ipod_alloc_buf() and initialise it with ipod_alloc_buf()
*/ */
/* Size of buffer for disk I/O */ /* Size of buffer for disk I/O - 8MB is large enough for any version
#define BUFFER_SIZE 6*1024*1024 of the Apple firmware, but not the Nano's RSRC image. */
#define BUFFER_SIZE 8*1024*1024
unsigned char* sectorbuf; unsigned char* sectorbuf;
char* get_parttype(int pt) char* get_parttype(int pt)
@ -101,13 +102,20 @@ struct partinfo_t {
unsigned char type; unsigned char type;
}; };
int static inline getint32le(unsigned char* buf) int static inline le2int(unsigned char* buf)
{ {
int32_t res = (buf[3] << 24) | (buf[2] << 16) | (buf[1] << 8) | buf[0]; int32_t res = (buf[3] << 24) | (buf[2] << 16) | (buf[1] << 8) | buf[0];
return res; return res;
} }
int static inline be2int(unsigned char* buf)
{
int32_t res = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
return res;
}
int static inline getint16le(char* buf) int static inline getint16le(char* buf)
{ {
int16_t res = (buf[1] << 8) | buf[0]; int16_t res = (buf[1] << 8) | buf[0];
@ -115,6 +123,27 @@ int static inline getint16le(char* buf)
return res; return res;
} }
void static inline short2le(unsigned short val, unsigned char* addr)
{
addr[0] = val & 0xFF;
addr[1] = (val >> 8) & 0xff;
}
void static inline int2le(unsigned int val, unsigned char* addr)
{
addr[0] = val & 0xFF;
addr[1] = (val >> 8) & 0xff;
addr[2] = (val >> 16) & 0xff;
addr[3] = (val >> 24) & 0xff;
}
void int2be(unsigned int val, unsigned char* addr)
{
addr[0] = (val >> 24) & 0xff;
addr[1] = (val >> 16) & 0xff;
addr[2] = (val >> 8) & 0xff;
addr[3] = val & 0xFF;
}
#define BYTES2INT32(array,pos)\ #define BYTES2INT32(array,pos)\
@ -157,13 +186,13 @@ int read_partinfo(HANDLE dh, int sector_size, struct partinfo_t* pinfo)
/* check that the boot sector is initialized */ /* check that the boot sector is initialized */
if ( (sector[510] != 0x55) || if ( (sector[510] != 0x55) ||
(sector[511] != 0xaa)) { (sector[511] != 0xaa)) {
fprintf(stderr,"Bad boot sector signature\n"); fprintf(stderr,"[ERR] Bad boot sector signature\n");
return 0; return -1;
} }
if ((memcmp(&sector[71],"iPod",4) != 0) && if ((memcmp(&sector[71],"iPod",4) != 0) &&
(memcmp(&sector[0x40],"This is your Apple iPod. You probably do not want to boot from it!",66) != 0) ) { (memcmp(&sector[0x40],"This is your Apple iPod. You probably do not want to boot from it!",66) != 0) ) {
fprintf(stderr,"Drive is not an iPod, aborting\n"); fprintf(stderr,"[ERR] Drive is not an iPod, aborting\n");
return -1; return -1;
} }
@ -183,8 +212,8 @@ int read_partinfo(HANDLE dh, int sector_size, struct partinfo_t* pinfo)
return 0; return 0;
} }
int disk_read(HANDLE dh, int outfile,unsigned long start, unsigned long count, int read_partition(HANDLE dh, int outfile,unsigned long start,
int sector_size) unsigned long count, int sector_size)
{ {
int res; int res;
unsigned long n; unsigned long n;
@ -214,7 +243,9 @@ int disk_read(HANDLE dh, int outfile,unsigned long start, unsigned long count,
} }
if (n < chunksize) { if (n < chunksize) {
fprintf(stderr,"[ERR] Short read in disk_read() - requested %d, got %lu\n",chunksize,n); fprintf(stderr,
"[ERR] Short read in disk_read() - requested %d, got %lu\n",
chunksize,n);
return -1; return -1;
} }
@ -228,7 +259,8 @@ int disk_read(HANDLE dh, int outfile,unsigned long start, unsigned long count,
} }
if (res != n) { if (res != n) {
fprintf(stderr,"Short write - requested %lu, received %d - aborting.\n",n,res); fprintf(stderr,
"Short write - requested %lu, received %d - aborting.\n",n,res);
return -1; return -1;
} }
} }
@ -237,7 +269,7 @@ int disk_read(HANDLE dh, int outfile,unsigned long start, unsigned long count,
return 0; return 0;
} }
int disk_write(HANDLE dh, int infile,unsigned long start, int sector_size) int write_partition(HANDLE dh, int infile,unsigned long start, int sector_size)
{ {
unsigned long res; unsigned long res;
int n; int n;
@ -281,14 +313,15 @@ int disk_write(HANDLE dh, int infile,unsigned long start, int sector_size)
} }
if (res != n) { if (res != n) {
fprintf(stderr,"Short write - requested %d, received %lu - aborting.\n",n,res); fprintf(stderr,"[ERR] Short write - requested %d, received %lu - aborting.\n",n,res);
return -1; return -1;
} }
byteswritten += res; byteswritten += res;
} }
fprintf(stderr,"[INFO] Wrote %d bytes plus %d bytes padding.\n",byteswritten-padding,padding); fprintf(stderr,"[INFO] Wrote %d bytes plus %d bytes padding.\n",
byteswritten-padding,padding);
return 0; return 0;
} }
@ -300,15 +333,16 @@ void print_usage(void) {
fprintf(stderr,"Usage: ipodpatcher device [action]\n"); fprintf(stderr,"Usage: ipodpatcher device [action]\n");
#endif #endif
fprintf(stderr,"\n"); fprintf(stderr,"\n");
fprintf(stderr,"Where [action] is one of:\n"); fprintf(stderr,"Where [action] is one of the following options:\n");
#if 0 fprintf(stderr," -l, --list\n");
fprintf(stderr," -e --extract-firmware filename.bin - extract firmware to a file\n"); fprintf(stderr," -r, --read-partition bootpartition.bin\n");
fprintf(stderr," -i --insert-firmware filename.bin - replace the firmware with the file\n"); fprintf(stderr," -w, --write-partition bootpartition.bin\n");
fprintf(stderr," -a --add-bootloader filename.bin - add a bootloader\n"); fprintf(stderr," -ef, --extract-firmware filename.ipod\n");
fprintf(stderr," -r --remove-bootloader - remove a bootloader\n"); fprintf(stderr," -rf, --replace-firmware filename.ipod\n");
#endif fprintf(stderr," -a, --add-bootloader filename.ipod\n");
fprintf(stderr," -l --list - list images in firmware partition\n"); fprintf(stderr," -d, --delete-bootloader\n");
fprintf(stderr,"\n"); fprintf(stderr,"\n");
#ifdef __WIN32__ #ifdef __WIN32__
fprintf(stderr,"DISKNO is the number (e.g. 2) Windows has assigned to your ipod's hard disk.\n"); fprintf(stderr,"DISKNO is the number (e.g. 2) Windows has assigned to your ipod's hard disk.\n");
fprintf(stderr,"The first hard disk in your computer (i.e. C:\\) will be disk0, the next disk\n"); fprintf(stderr,"The first hard disk in your computer (i.e. C:\\) will be disk0, the next disk\n");
@ -325,7 +359,13 @@ void print_usage(void) {
enum { enum {
NONE, NONE,
SHOW_INFO, SHOW_INFO,
LIST_IMAGES LIST_IMAGES,
REMOVE_BOOTLOADER,
INSERT_BOOTLOADER,
EXTRACT_FIRMWARE,
REPLACE_FIRMWARE,
READ_PARTITION,
WRITE_PARTITION
}; };
char* ftypename[] = { "OSOS", "RSRC", "AUPD", "HIBE" }; char* ftypename[] = { "OSOS", "RSRC", "AUPD", "HIBE" };
@ -340,7 +380,7 @@ enum firmwaretype_t {
struct ipod_directory_t { struct ipod_directory_t {
enum firmwaretype_t ftype; enum firmwaretype_t ftype;
int id; int id;
uint32_t devOffset; uint32_t devOffset; /* Offset of image relative to one sector into bootpart*/
uint32_t len; uint32_t len;
uint32_t addr; uint32_t addr;
uint32_t entryOffset; uint32_t entryOffset;
@ -349,14 +389,209 @@ struct ipod_directory_t {
uint32_t loadAddr; uint32_t loadAddr;
}; };
int read_directory(HANDLE dh, int start, int sector_size, struct ipod_directory_t* ipod_directory) int remove_bootloader(HANDLE dh, int start, int sector_size,
struct ipod_directory_t* ipod_directory)
{
fprintf(stderr,"[ERR] Sorry, not yet implemented.\n");
return -1;
}
int replace_firmware(HANDLE dh, char* filename, int start, int sector_size,
int nimages, struct ipod_directory_t* ipod_directory,
off_t diroffset, int modelnum, char* modelname)
{
int length;
int i;
int n;
int infile;
int newsize;
int bytesavailable;
unsigned long chksum=0;
unsigned long filechksum=0;
unsigned char header[8]; /* Header for .ipod file */
/* First check that the input file is the correct type for this ipod. */
infile=open(filename,O_RDONLY);
if (infile < 0) {
fprintf(stderr,"[ERR] Couldn't open input file %s\n",filename);
}
n = read(infile,header,8);
if (n < 8) {
fprintf(stderr,"[ERR] Failed to read header from %s\n",filename);
}
if (memcmp(header+4,modelname,4)!=0) {
fprintf(stderr,"[ERR] Model name in input file (%c%c%c%c) doesn't match ipod model (%s)\n",
header[4],header[5],header[6],header[7],modelname);
close(infile);
return -1;
}
filechksum = be2int(header);
length=filesize(infile)-8;
newsize=(length+sector_size-1)&~(sector_size-1);
fprintf(stderr,"[INFO] Padding input file from 0x%08x to 0x%08x bytes\n",
length,newsize);
if (newsize > BUFFER_SIZE) {
fprintf(stderr,"[ERR] Input file too big for buffer\n");
close(infile);
return -1;
}
/* Check if we have enough space */
/* TODO: Check the size of the partition. */
if (nimages > 1) {
bytesavailable=ipod_directory[1].devOffset-ipod_directory[0].devOffset;
if (bytesavailable < newsize) {
fprintf(stderr,"[INFO] Moving images to create room for new firmware...\n");
/* TODO: Implement image movement */
fprintf(stderr,"[ERR] Image movement not yet implemented.\n");
close(infile);
return -1;
}
}
fprintf(stderr,"[INFO] Reading input file...\n");
/* We now know we have enough space, so write it. */
memset(sectorbuf+length,0,newsize-length);
n = read(infile,sectorbuf,length);
if (n < 0) {
fprintf(stderr,"[ERR] Couldn't read input file\n");
close(infile);
return -1;
}
close(infile);
chksum = modelnum;
for (i = 0; i < length; i++) {
/* add 8 unsigned bits but keep a 32 bit sum */
chksum += sectorbuf[i];
}
if (chksum == filechksum) {
fprintf(stderr,"[INFO] Checksum OK in %s\n",filename);
} else {
fprintf(stderr,"[ERR] Checksum in %s failed check\n",filename);
return -1;
}
if (ipod_seek(dh,start+sector_size+ipod_directory[0].devOffset) < 0) {
fprintf(stderr,"[ERR] Seek failed\n");
return -1;
}
if ((n = ipod_write(dh,sectorbuf,newsize)) < 0) {
perror("[ERR] Write failed\n");
return -1;
}
if (n < newsize) {
fprintf(stderr,"[ERR] Short write - requested %d bytes, received %d\n"
,i,n);
return -1;
}
fprintf(stderr,"[INFO] Wrote %d bytes to firmware partition\n",n);
/* Now we need to update the "len", "entryOffset" and "chksum" fields */
chksum = 0;
for (i = 0; i < length; i++) {
/* add 8 unsigned bits but keep a 32 bit sum */
chksum += sectorbuf[i];
}
/* Read directory */
if (ipod_seek(dh,start + diroffset) < 0) { return -1; }
n=ipod_read(dh, sectorbuf, sector_size);
if (n < 0) { return -1; }
/* Update entries for image 0 */
int2le(length,sectorbuf+16);
int2le(0,sectorbuf+24);
int2le(chksum,sectorbuf+28);
/* Write directory */
if (ipod_seek(dh,start + diroffset) < 0) { return -1; }
n=ipod_write(dh, sectorbuf, sector_size);
if (n < 0) { return -1; }
return 0;
}
int extract_firmware(HANDLE dh, char* filename, int start, int sector_size,
struct ipod_directory_t* ipod_directory,
int modelnum, char* modelname)
{
int length;
int i;
int outfile;
int n;
unsigned long chksum=0; /* 32 bit checksum - Rockbox .ipod style*/
unsigned char header[8]; /* Header for .ipod file */
if (ipod_directory[0].entryOffset != 0) {
/* We have a bootloader... */
length = ipod_directory[0].entryOffset;
} else {
length = ipod_directory[0].len;
}
fprintf(stderr,"[INFO] Reading firmware (%d bytes)\n",length);
if (ipod_seek(dh,start+sector_size+ipod_directory[0].devOffset) < 0) {
return -1;
}
i = (length+sector_size-1) & ~(sector_size-1);
fprintf(stderr,"[INFO] Padding read from 0x%08x to 0x%08x bytes\n",
length,i);
if ((n = ipod_read(dh,sectorbuf,i)) < 0) {
return -1;
}
if (n < i) {
fprintf(stderr,"[ERR] Short read - requested %d bytes, received %d\n",
i,n);
return -1;
}
chksum = modelnum;
for (i = 0; i < length; i++) {
/* add 8 unsigned bits but keep a 32 bit sum */
chksum += sectorbuf[i];
}
int2be(chksum,header);
memcpy(header+4,modelname,4);
outfile = open(filename,O_CREAT|O_WRONLY|O_BINARY,0666);
if (outfile < 0) {
fprintf(stderr,"[ERR] Couldn't open file %s\n",filename);
return -1;
}
write(outfile,header,8);
write(outfile,sectorbuf,ipod_directory[0].len);
close(outfile);
return 0;
}
int read_directory(HANDLE dh, int start, int sector_size,
struct ipod_directory_t* ipod_directory, off_t* diroffset)
{ {
int n; int n;
int nimages; int nimages;
off_t diroffset;
unsigned char* p; unsigned char* p;
/* Read firmware partition header (first 512 bytes of disk - but let's read a whole sector) */ /* Read firmware partition header (first 512 bytes of disk - but
let's read a whole sector) */
if (ipod_seek(dh, start) < 0) { return -1; } if (ipod_seek(dh, start) < 0) { return -1; }
@ -369,14 +604,14 @@ int read_directory(HANDLE dh, int start, int sector_size, struct ipod_directory_
} }
if (memcmp(sectorbuf+0x100,"]ih[",4)!=0) { if (memcmp(sectorbuf+0x100,"]ih[",4)!=0) {
fprintf(stderr,"[ERR] Bad firmware directory\n"); fprintf(stderr,"[ERR] Bad firmware directory\n");
return -1; return -1;
} }
diroffset=getint32le(sectorbuf+0x104) + 0x200; *diroffset=le2int(sectorbuf+0x104) + 0x200;
/* Read directory */ /* Read directory */
if (ipod_seek(dh,start + diroffset) < 0) { return -1; } if (ipod_seek(dh,start + *diroffset) < 0) { return -1; }
n=ipod_read(dh, sectorbuf, sector_size); n=ipod_read(dh, sectorbuf, sector_size);
if (n < 0) { return -1; } if (n < 0) { return -1; }
@ -384,7 +619,8 @@ int read_directory(HANDLE dh, int start, int sector_size, struct ipod_directory_
nimages=0; nimages=0;
p = sectorbuf; p = sectorbuf;
while ((nimages < MAX_IMAGES) && (p < (sectorbuf + 400)) && (memcmp(p,"!ATA",4)==0)) { while ((nimages < MAX_IMAGES) && (p < (sectorbuf + 400)) &&
(memcmp(p,"!ATA",4)==0)) {
p+=4; p+=4;
if (memcmp(p,"soso",4)==0) { if (memcmp(p,"soso",4)==0) {
ipod_directory[nimages].ftype=FTYPE_OSOS; ipod_directory[nimages].ftype=FTYPE_OSOS;
@ -395,38 +631,40 @@ int read_directory(HANDLE dh, int start, int sector_size, struct ipod_directory_
} else if (memcmp(p,"ebih",4)==0) { } else if (memcmp(p,"ebih",4)==0) {
ipod_directory[nimages].ftype=FTYPE_HIBE; ipod_directory[nimages].ftype=FTYPE_HIBE;
} else { } else {
fprintf(stderr,"[ERR] Unknown image type %c%c%c%c\n",p[0],p[1],p[2],p[3]); fprintf(stderr,"[ERR] Unknown image type %c%c%c%c\n",
p[0],p[1],p[2],p[3]);
} }
p+=4; p+=4;
ipod_directory[nimages].id=getint32le(p); ipod_directory[nimages].id=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].devOffset=getint32le(p); ipod_directory[nimages].devOffset=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].len=getint32le(p); ipod_directory[nimages].len=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].addr=getint32le(p); ipod_directory[nimages].addr=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].entryOffset=getint32le(p); ipod_directory[nimages].entryOffset=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].chksum=getint32le(p); ipod_directory[nimages].chksum=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].vers=getint32le(p); ipod_directory[nimages].vers=le2int(p);
p+=4; p+=4;
ipod_directory[nimages].loadAddr=getint32le(p); ipod_directory[nimages].loadAddr=le2int(p);
p+=4; p+=4;
nimages++; nimages++;
} }
return nimages; return nimages;
} }
int list_images(int nimages, struct ipod_directory_t* ipod_directory) int list_images(int nimages, struct ipod_directory_t* ipod_directory,
int sector_size)
{ {
int i; int i;
#ifdef DEBUG #ifdef DEBUG
printf(" Type id devOffset len addr entryOffset chksum vers loadAddr\n"); printf(" Type id devOffset len addr entryOffset chksum vers loadAddr devOffset+len\n");
for (i = 0 ; i < nimages; i++) { for (i = 0 ; i < nimages; i++) {
printf("%d - %s 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x\n",i, printf("%d - %s 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x\n",i,
ftypename[ipod_directory[i].ftype], ftypename[ipod_directory[i].ftype],
ipod_directory[i].id, ipod_directory[i].id,
ipod_directory[i].devOffset, ipod_directory[i].devOffset,
@ -435,7 +673,8 @@ int list_images(int nimages, struct ipod_directory_t* ipod_directory)
ipod_directory[i].entryOffset, ipod_directory[i].entryOffset,
ipod_directory[i].chksum, ipod_directory[i].chksum,
ipod_directory[i].vers, ipod_directory[i].vers,
ipod_directory[i].loadAddr); ipod_directory[i].loadAddr,
ipod_directory[i].devOffset+sector_size+((ipod_directory[i].len+sector_size-1)&~(sector_size-1)));
} }
#endif #endif
@ -448,14 +687,19 @@ int list_images(int nimages, struct ipod_directory_t* ipod_directory)
switch(ipod_directory[i].ftype) { switch(ipod_directory[i].ftype) {
case FTYPE_OSOS: case FTYPE_OSOS:
if (ipod_directory[i].entryOffset==0) { if (ipod_directory[i].entryOffset==0) {
printf(" Main firmware - %d bytes\n",ipod_directory[i].len); printf(" Main firmware - %d bytes\n",
ipod_directory[i].len);
} else { } else {
printf(" Main firmware - %d bytes\n",ipod_directory[i].entryOffset); printf(" Main firmware - %d bytes\n",
printf(" Third-party bootloader - %d bytes\n",ipod_directory[i].len-ipod_directory[i].entryOffset); ipod_directory[i].entryOffset);
printf(" Third-party bootloader - %d bytes\n",
ipod_directory[i].len-ipod_directory[i].entryOffset);
} }
break; break;
default: default:
printf(" %s - %d bytes\n",ftypename[ipod_directory[i].ftype],ipod_directory[i].len); printf(" %s - %d bytes\n",
ftypename[ipod_directory[i].ftype],
ipod_directory[i].len);
} }
} }
printf("\n"); printf("\n");
@ -463,12 +707,21 @@ int list_images(int nimages, struct ipod_directory_t* ipod_directory)
return 0; return 0;
} }
int main(int argc, char* argv[]) int main(int argc, char* argv[])
{ {
int i; int i;
int infile, outfile;
int ipod_version; int ipod_version;
unsigned int inputsize;
struct partinfo_t pinfo[4]; /* space for 4 partitions on 1 drive */ struct partinfo_t pinfo[4]; /* space for 4 partitions on 1 drive */
int nimages; int nimages;
off_t diroffset;
char* modelname;
int modelnum;
char* filename;
struct ipod_directory_t ipod_directory[MAX_IMAGES]; struct ipod_directory_t ipod_directory[MAX_IMAGES];
int action = SHOW_INFO; int action = SHOW_INFO;
int sector_size; int sector_size;
@ -498,6 +751,37 @@ int main(int argc, char* argv[])
if ((strcmp(argv[i],"-l")==0) || (strcmp(argv[i],"--list")==0)) { if ((strcmp(argv[i],"-l")==0) || (strcmp(argv[i],"--list")==0)) {
action = LIST_IMAGES; action = LIST_IMAGES;
i++; i++;
} else if (strcmp(argv[i],"--remove-bootloader")==0) {
action = REMOVE_BOOTLOADER;
i++;
} else if ((strcmp(argv[i],"-ef")==0) ||
(strcmp(argv[i],"--extract-firmware")==0)) {
action = EXTRACT_FIRMWARE;
i++;
if (i == argc) { print_usage(); return 1; }
filename=argv[i];
i++;
} else if ((strcmp(argv[i],"-rf")==0) ||
(strcmp(argv[i],"--replace-firmware")==0)) {
action = REPLACE_FIRMWARE;
i++;
if (i == argc) { print_usage(); return 1; }
filename=argv[i];
i++;
} else if ((strcmp(argv[i],"-r")==0) ||
(strcmp(argv[i],"--read-partition")==0)) {
action = READ_PARTITION;
i++;
if (i == argc) { print_usage(); return 1; }
filename=argv[i];
i++;
} else if ((strcmp(argv[i],"-w")==0) ||
(strcmp(argv[i],"--write-partition")==0)) {
action = WRITE_PARTITION;
i++;
if (i == argc) { print_usage(); return 1; }
filename=argv[i];
i++;
} else { } else {
print_usage(); return 1; print_usage(); return 1;
} }
@ -531,7 +815,8 @@ int main(int argc, char* argv[])
return 3; return 3;
} }
nimages=read_directory(dh, pinfo[0].start*sector_size, sector_size, ipod_directory); nimages=read_directory(dh, pinfo[0].start*sector_size, sector_size,
ipod_directory, &diroffset);
if (nimages <= 0) { if (nimages <= 0) {
fprintf(stderr,"[ERR] Failed to read firmware directory\n"); fprintf(stderr,"[ERR] Failed to read firmware directory\n");
return 1; return 1;
@ -540,30 +825,93 @@ int main(int argc, char* argv[])
ipod_version=(ipod_directory[0].vers>>12) & 0x0f; ipod_version=(ipod_directory[0].vers>>12) & 0x0f;
printf("[INFO] Ipod model: "); printf("[INFO] Ipod model: ");
switch (ipod_version) { switch (ipod_version) {
case 0x3: printf("3rd Generation\n"); break; case 0x3:
case 0x4: printf("1st Generation Mini\n"); break; printf("3rd Generation\n");
case 0x5: printf("4th Generation\n"); break; modelnum = 7;
case 0x6: printf("Photo/Color\n"); break; modelname = "ip3g";
case 0x7: printf("2nd Generation Mini\n"); break; break;
case 0xc: printf("1st Generation Nano\n"); break; case 0x4:
case 0xb: printf("Video (aka 5th Generation)\n"); break; printf("1st Generation Mini\n");
default: printf("UNKNOWN (Firmware version is %08x)\n",ipod_directory[0].vers); modelnum = 9;
modelname = "mini";
break;
case 0x5:
printf("4th Generation\n");
modelnum = 8;
modelname = "ip4g";
break;
case 0x6:
printf("Photo/Color\n");
modelnum = 3;
modelname = "ipco";
break;
case 0x7:
printf("2nd Generation Mini\n");
modelnum = 11;
modelname = "mn2g";
break;
case 0xc:
printf("1st Generation Nano\n");
modelnum = 4;
modelname = "nano";
break;
case 0xb:
printf("Video (aka 5th Generation)\n");
modelnum = 5;
modelname = "ipvd";
break;
default:
printf("[ERR] Unknown firmware version (0x%08x)\n",
ipod_directory[0].vers);
return -1;
} }
if (action==LIST_IMAGES) { if (action==LIST_IMAGES) {
list_images(nimages,ipod_directory); list_images(nimages,ipod_directory,sector_size);
#if 0 } else if (action==REMOVE_BOOTLOADER) {
} else if (mode==READ) { if (ipod_directory[0].entryOffset==0) {
fprintf(stderr,"[ERR] No bootloader detected.\n");
} else {
if (remove_bootloader(dh, pinfo[0].start*sector_size, sector_size,
ipod_directory)==0) {
fprintf(stderr,"[INFO] Bootloader removed.\n");
}
}
} else if (action==REPLACE_FIRMWARE) {
if (ipod_reopen_rw(&dh, devicename) < 0) {
return 5;
}
if (replace_firmware(dh, filename,pinfo[0].start*sector_size,
sector_size, nimages, ipod_directory, diroffset,
modelnum, modelname)==0) {
fprintf(stderr,"[INFO] Firmware replaced with %s.\n",filename);
} else {
fprintf(stderr,"[ERR] --replace-firmware failed.\n");
}
} else if (action==EXTRACT_FIRMWARE) {
if (extract_firmware(dh, filename,pinfo[0].start*sector_size,
sector_size, ipod_directory, modelnum, modelname
)==0) {
fprintf(stderr,"[INFO] Firmware extracted to %s.\n",filename);
} else {
fprintf(stderr,"[ERR] --extract-firmware failed.\n");
}
} else if (action==READ_PARTITION) {
outfile = open(filename,O_CREAT|O_WRONLY|O_BINARY,S_IREAD|S_IWRITE); outfile = open(filename,O_CREAT|O_WRONLY|O_BINARY,S_IREAD|S_IWRITE);
if (outfile < 0) { if (outfile < 0) {
perror(filename); perror(filename);
return 4; return 4;
} }
res = disk_read(dh,outfile,pinfo[p].start,pinfo[p].size,sector_size); if (read_partition(dh, outfile, pinfo[0].start, pinfo[0].size,
sector_size) < 0) {
fprintf(stderr,"[ERR] --read-partition failed.\n");
} else {
fprintf(stderr,"[INFO] Partition extracted to %s.\n",filename);
}
close(outfile); close(outfile);
} else if (mode==WRITE) { } else if (action==WRITE_PARTITION) {
if (ipod_reopen_rw(&dh, devicename) < 0) { if (ipod_reopen_rw(&dh, devicename) < 0) {
return 5; return 5;
} }
@ -577,16 +925,20 @@ int main(int argc, char* argv[])
/* Check filesize is <= partition size */ /* Check filesize is <= partition size */
inputsize=filesize(infile); inputsize=filesize(infile);
if (inputsize > 0) { if (inputsize > 0) {
if (inputsize <= (pinfo[p].size*sector_size)) { if (inputsize <= (pinfo[0].size*sector_size)) {
fprintf(stderr,"[INFO] Input file is %lu bytes\n",inputsize); fprintf(stderr,"[INFO] Input file is %u bytes\n",inputsize);
res = disk_write(dh,infile,pinfo[p].start,sector_size); if (write_partition(dh,infile,pinfo[0].start,
sector_size) < 0) {
fprintf(stderr,"[ERR] --write-partition failed.\n");
} else {
fprintf(stderr,"[INFO] %s restored to partition\n",filename);
}
} else { } else {
fprintf(stderr,"[ERR] File is too large for firmware partition, aborting.\n"); fprintf(stderr,"[ERR] File is too large for firmware partition, aborting.\n");
} }
} }
close(infile); close(infile);
#endif
} }
ipod_close(dh); ipod_close(dh);