1 /***************************************************************************
3 * Open \______ \ ____ ____ | | _\_ |__ _______ ___
4 * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
5 * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
6 * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
10 * Copyright (C) 2006-2007 Dave Chapman
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License
14 * as published by the Free Software Foundation; either version 2
15 * of the License, or (at your option) any later version.
17 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
18 * KIND, either express or implied.
20 ****************************************************************************/
28 #include <sys/types.h>
31 #include "ipodpatcher.h"
36 #define VERSION "5.0 with v4.0 bootloaders (v1.0 for 2nd Gen Nano)"
61 void print_macpod_warning(void)
63 printf("[INFO] ************************************************************************\n");
64 printf("[INFO] *** WARNING FOR ROCKBOX USERS\n");
65 printf("[INFO] *** You must convert this ipod to FAT32 format (aka a \"winpod\")\n");
66 printf("[INFO] *** if you want to run Rockbox. Rockbox WILL NOT work on this ipod.\n");
67 printf("[INFO] *** See http://www.rockbox.org/twiki/bin/view/Main/IpodConversionToFAT32\n");
68 printf("[INFO] ************************************************************************\n");
71 void print_usage(void)
73 fprintf(stderr
,"Usage: ipodpatcher --scan\n");
75 fprintf(stderr
," or ipodpatcher [DISKNO] [action]\n");
77 fprintf(stderr
," or ipodpatcher [device] [action]\n");
80 fprintf(stderr
,"Where [action] is one of the following options:\n");
82 fprintf(stderr
," --install\n");
84 fprintf(stderr
," -l, --list\n");
85 fprintf(stderr
," -r, --read-partition bootpartition.bin\n");
86 fprintf(stderr
," -w, --write-partition bootpartition.bin\n");
87 fprintf(stderr
," -rf, --read-firmware filename.ipod[x]\n");
88 fprintf(stderr
," -rfb, --read-firmware-bin filename.bin\n");
89 fprintf(stderr
," -wf, --write-firmware filename.ipod[x]\n");
90 fprintf(stderr
," -wfb, --write-firmware-bin filename.bin\n");
92 fprintf(stderr
," -we, --write-embedded\n");
94 fprintf(stderr
," -a, --add-bootloader filename.ipod[x]\n");
95 fprintf(stderr
," -ab, --add-bootloader-bin filename.bin\n");
96 fprintf(stderr
," -d, --delete-bootloader\n");
97 fprintf(stderr
," -f, --format\n");
98 fprintf(stderr
," -c, --convert\n");
99 fprintf(stderr
," --read-aupd filename.bin\n");
100 fprintf(stderr
," --write-aupd filename.bin\n");
101 fprintf(stderr
," -x --dump-xml filename.xml\n");
102 fprintf(stderr
,"\n");
104 fprintf(stderr
,"The .ipodx extension is used for encrypted images for the 2nd Gen Nano.\n\n");
107 fprintf(stderr
,"DISKNO is the number (e.g. 2) Windows has assigned to your ipod's hard disk.\n");
108 fprintf(stderr
,"The first hard disk in your computer (i.e. C:\\) will be disk 0, the next disk\n");
109 fprintf(stderr
,"will be disk 1 etc. ipodpatcher will refuse to access a disk unless it\n");
110 fprintf(stderr
,"can identify it as being an ipod.\n");
111 fprintf(stderr
,"\n");
113 #if defined(linux) || defined (__linux)
114 fprintf(stderr
,"\"device\" is the device node (e.g. /dev/sda) assigned to your ipod.\n");
115 #elif defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)
116 fprintf(stderr
,"\"device\" is the device node (e.g. /dev/da1) assigned to your ipod.\n");
117 #elif defined(__APPLE__) && defined(__MACH__)
118 fprintf(stderr
,"\"device\" is the device node (e.g. /dev/disk1) assigned to your ipod.\n");
120 fprintf(stderr
,"ipodpatcher will refuse to access a disk unless it can identify it as being\n");
121 fprintf(stderr
,"an ipod.\n");
125 void display_partinfo(struct ipod_t
* ipod
)
128 double sectors_per_MB
= (1024.0*1024.0)/ipod
->sector_size
;
130 printf("[INFO] Part Start Sector End Sector Size (MB) Type\n");
131 for ( i
= 0; i
< 4; i
++ ) {
132 if (ipod
->pinfo
[i
].start
!= 0) {
133 printf("[INFO] %d %10ld %10ld %10.1f %s (0x%02x)\n",
135 (long int)ipod
->pinfo
[i
].start
,
136 (long int)ipod
->pinfo
[i
].start
+ipod
->pinfo
[i
].size
-1,
137 ipod
->pinfo
[i
].size
/sectors_per_MB
,
138 get_parttype(ipod
->pinfo
[i
].type
),
139 (int)ipod
->pinfo
[i
].type
);
145 int main(int argc
, char* argv
[])
151 unsigned int inputsize
;
153 int action
= SHOW_INFO
;
157 fprintf(stderr
,"ipodpatcher " VERSION
"\n");
158 fprintf(stderr
,"(C) Dave Chapman 2006-2009\n");
159 fprintf(stderr
,"This is free software; see the source for copying conditions. There is NO\n");
160 fprintf(stderr
,"warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\n");
162 if ((argc
> 1) && ((strcmp(argv
[1],"-h")==0) || (strcmp(argv
[1],"--help")==0))) {
167 if (ipod_alloc_buffer(&ipod_sectorbuf
,BUFFER_SIZE
) < 0) {
168 fprintf(stderr
,"Failed to allocate memory buffer\n");
171 if ((argc
> 1) && (strcmp(argv
[1],"--scan")==0)) {
172 if (ipod_scan(&ipod
) == 0)
173 fprintf(stderr
,"[ERR] No ipods found.\n");
177 /* If the first parameter doesn't start with -, then we interpret it as a device */
178 if ((argc
> 1) && (argv
[1][0] != '-')) {
181 snprintf(ipod
.diskname
,sizeof(ipod
.diskname
),"\\\\.\\PhysicalDrive%s",argv
[1]);
183 strncpy(ipod
.diskname
,argv
[1],sizeof(ipod
.diskname
));
187 /* Autoscan for ipods */
188 n
= ipod_scan(&ipod
);
190 fprintf(stderr
,"[ERR] No ipods found, aborting\n");
191 fprintf(stderr
,"[ERR] Please connect your ipod and ensure it is in disk mode\n");
192 #if defined(__APPLE__) && defined(__MACH__)
193 fprintf(stderr
,"[ERR] Also ensure that itunes is closed, and that your ipod is not mounted.\n");
194 #elif !defined(__WIN32__)
196 fprintf(stderr
,"[ERR] You may also need to run ipodpatcher as root.\n");
199 fprintf(stderr
,"[ERR] Please refer to the Rockbox manual if you continue to have problems.\n");
201 fprintf(stderr
,"[ERR] %d ipods found, aborting\n",n
);
202 fprintf(stderr
,"[ERR] Please connect only one ipod and re-run ipodpatcher.\n");
208 printf("\nPress ENTER to exit ipodpatcher :");
209 fgets(yesno
,4,stdin
);
219 action
= INTERACTIVE
;
225 if ((strcmp(argv
[i
],"-l")==0) || (strcmp(argv
[i
],"--list")==0)) {
226 action
= LIST_IMAGES
;
229 } else if (strcmp(argv
[i
],"--install")==0) {
233 } else if ((strcmp(argv
[i
],"-d")==0) ||
234 (strcmp(argv
[i
],"--delete-bootloader")==0)) {
235 action
= DELETE_BOOTLOADER
;
237 } else if ((strcmp(argv
[i
],"-a")==0) ||
238 (strcmp(argv
[i
],"--add-bootloader")==0)) {
239 action
= ADD_BOOTLOADER
;
240 type
= FILETYPE_DOT_IPOD
;
242 if (i
== argc
) { print_usage(); return 1; }
245 } else if ((strcmp(argv
[i
],"-ab")==0) ||
246 (strcmp(argv
[i
],"--add-bootloader-bin")==0)) {
247 action
= ADD_BOOTLOADER
;
248 type
= FILETYPE_DOT_BIN
;
250 if (i
== argc
) { print_usage(); return 1; }
253 } else if ((strcmp(argv
[i
],"-rf")==0) ||
254 (strcmp(argv
[i
],"--read-firmware")==0)) {
255 action
= READ_FIRMWARE
;
256 type
= FILETYPE_DOT_IPOD
;
258 if (i
== argc
) { print_usage(); return 1; }
261 } else if ((strcmp(argv
[i
],"-rfb")==0) ||
262 (strcmp(argv
[i
],"--read-firmware-bin")==0)) {
263 action
= READ_FIRMWARE
;
264 type
= FILETYPE_DOT_BIN
;
266 if (i
== argc
) { print_usage(); return 1; }
270 } else if ((strcmp(argv
[i
],"-we")==0) ||
271 (strcmp(argv
[i
],"--write-embedded")==0)) {
272 action
= WRITE_FIRMWARE
;
273 type
= FILETYPE_INTERNAL
;
274 filename
="[embedded bootloader]"; /* Only displayed for user */
277 } else if ((strcmp(argv
[i
],"-wf")==0) ||
278 (strcmp(argv
[i
],"--write-firmware")==0)) {
279 action
= WRITE_FIRMWARE
;
280 type
= FILETYPE_DOT_IPOD
;
282 if (i
== argc
) { print_usage(); return 1; }
285 } else if ((strcmp(argv
[i
],"-wfb")==0) ||
286 (strcmp(argv
[i
],"--write-firmware-bin")==0)) {
287 action
= WRITE_FIRMWARE
;
288 type
= FILETYPE_DOT_BIN
;
290 if (i
== argc
) { print_usage(); return 1; }
293 } else if ((strcmp(argv
[i
],"-r")==0) ||
294 (strcmp(argv
[i
],"--read-partition")==0)) {
295 action
= READ_PARTITION
;
297 if (i
== argc
) { print_usage(); return 1; }
300 } else if ((strcmp(argv
[i
],"-w")==0) ||
301 (strcmp(argv
[i
],"--write-partition")==0)) {
302 action
= WRITE_PARTITION
;
304 if (i
== argc
) { print_usage(); return 1; }
307 } else if ((strcmp(argv
[i
],"-v")==0) ||
308 (strcmp(argv
[i
],"--verbose")==0)) {
311 } else if ((strcmp(argv
[i
],"-f")==0) ||
312 (strcmp(argv
[i
],"--format")==0)) {
313 action
= FORMAT_PARTITION
;
315 } else if (strcmp(argv
[i
],"--read-aupd")==0) {
318 if (i
== argc
) { print_usage(); return 1; }
321 } else if (strcmp(argv
[i
],"--write-aupd")==0) {
324 if (i
== argc
) { print_usage(); return 1; }
327 } else if ((strcmp(argv
[i
],"-x")==0) ||
328 (strcmp(argv
[i
],"--dump-xml")==0)) {
331 if (i
== argc
) { print_usage(); return 1; }
334 } else if ((strcmp(argv
[i
],"-c")==0) ||
335 (strcmp(argv
[i
],"--convert")==0)) {
336 action
= CONVERT_TO_FAT32
;
339 print_usage(); return 1;
343 if (ipod
.diskname
[0]==0) {
348 if (ipod_open(&ipod
, 0) < 0) {
352 fprintf(stderr
,"[INFO] Reading partition table from %s\n",ipod
.diskname
);
353 fprintf(stderr
,"[INFO] Sector size is %d bytes\n",ipod
.sector_size
);
355 if (read_partinfo(&ipod
,0) < 0) {
359 display_partinfo(&ipod
);
361 if (ipod
.pinfo
[0].start
==0) {
362 fprintf(stderr
,"[ERR] No partition 0 on disk:\n");
363 display_partinfo(&ipod
);
367 read_directory(&ipod
);
369 if (ipod
.nimages
<= 0) {
370 fprintf(stderr
,"[ERR] Failed to read firmware directory - nimages=%d\n",ipod
.nimages
);
374 if (getmodel(&ipod
,(ipod
.ipod_directory
[ipod
.ososimage
].vers
>>8)) < 0) {
375 fprintf(stderr
,"[ERR] Unknown version number in firmware (%08x)\n",
376 ipod
.ipod_directory
[ipod
.ososimage
].vers
);
381 /* Windows requires the ipod in R/W mode for SCSI Inquiry */
382 if (ipod_reopen_rw(&ipod
) < 0) {
388 /* Read the XML info, and if successful, look for the ramsize
389 (only available for some models - set to 0 if not known) */
393 if (ipod_get_xmlinfo(&ipod
) == 0) {
394 ipod_get_ramsize(&ipod
);
397 printf("[INFO] Ipod model: %s ",ipod
.modelstr
);
398 if (ipod
.ramsize
> 0) { printf("(%dMB RAM) ",ipod
.ramsize
); }
399 printf("(\"%s\")\n",ipod
.macpod
? "macpod" : "winpod");
402 print_macpod_warning();
405 if (action
==LIST_IMAGES
) {
408 } else if (action
==INTERACTIVE
) {
410 printf("Enter i to install the Rockbox bootloader, u to uninstall\n or c to cancel and do nothing (i/u/c) :");
412 if (fgets(yesno
,4,stdin
)) {
414 if (ipod_reopen_rw(&ipod
) < 0) {
418 if (add_bootloader(&ipod
, NULL
, FILETYPE_INTERNAL
)==0) {
419 fprintf(stderr
,"[INFO] Bootloader installed successfully.\n");
421 fprintf(stderr
,"[ERR] --install failed.\n");
423 } else if (yesno
[0]=='u') {
424 if (ipod_reopen_rw(&ipod
) < 0) {
428 if (delete_bootloader(&ipod
)==0) {
429 fprintf(stderr
,"[INFO] Bootloader removed.\n");
431 fprintf(stderr
,"[ERR] Bootloader removal failed.\n");
436 } else if (action
==DELETE_BOOTLOADER
) {
437 if (ipod_reopen_rw(&ipod
) < 0) {
441 if (ipod
.ipod_directory
[0].entryOffset
==0) {
442 fprintf(stderr
,"[ERR] No bootloader detected.\n");
444 if (delete_bootloader(&ipod
)==0) {
445 fprintf(stderr
,"[INFO] Bootloader removed.\n");
447 fprintf(stderr
,"[ERR] --delete-bootloader failed.\n");
450 } else if (action
==ADD_BOOTLOADER
) {
451 if (ipod_reopen_rw(&ipod
) < 0) {
455 if (add_bootloader(&ipod
, filename
, type
)==0) {
456 fprintf(stderr
,"[INFO] Bootloader %s written to device.\n",filename
);
458 fprintf(stderr
,"[ERR] --add-bootloader failed.\n");
461 } else if (action
==INSTALL
) {
462 if (ipod_reopen_rw(&ipod
) < 0) {
466 if (add_bootloader(&ipod
, NULL
, FILETYPE_INTERNAL
)==0) {
467 fprintf(stderr
,"[INFO] Bootloader installed successfully.\n");
469 fprintf(stderr
,"[ERR] --install failed.\n");
472 } else if (action
==WRITE_FIRMWARE
) {
473 if (ipod_reopen_rw(&ipod
) < 0) {
477 if (write_firmware(&ipod
, filename
,type
)==0) {
478 fprintf(stderr
,"[INFO] Firmware %s written to device.\n",filename
);
480 fprintf(stderr
,"[ERR] --write-firmware failed.\n");
482 } else if (action
==READ_FIRMWARE
) {
483 if (read_firmware(&ipod
, filename
, type
)==0) {
484 fprintf(stderr
,"[INFO] Firmware read to file %s.\n",filename
);
486 fprintf(stderr
,"[ERR] --read-firmware failed.\n");
488 } else if (action
==READ_AUPD
) {
489 if (read_aupd(&ipod
, filename
)==0) {
490 fprintf(stderr
,"[INFO] AUPD image read to file %s.\n",filename
);
492 fprintf(stderr
,"[ERR] --read-aupd failed.\n");
494 } else if (action
==WRITE_AUPD
) {
495 if (ipod_reopen_rw(&ipod
) < 0) {
499 if (write_aupd(&ipod
, filename
)==0) {
500 fprintf(stderr
,"[INFO] AUPD image %s written to device.\n",filename
);
502 fprintf(stderr
,"[ERR] --write-aupd failed.\n");
504 } else if (action
==DUMP_XML
) {
505 if (ipod
.xmlinfo
== NULL
) {
506 fprintf(stderr
,"[ERR] No XML to write\n");
510 outfile
= open(filename
,O_CREAT
|O_TRUNC
|O_WRONLY
|O_BINARY
,S_IREAD
|S_IWRITE
);
516 if (write(outfile
, ipod
.xmlinfo
, ipod
.xmlinfo_len
) < 0) {
517 fprintf(stderr
,"[ERR] --dump-xml failed.\n");
519 fprintf(stderr
,"[INFO] XML info written to %s.\n",filename
);
522 } else if (action
==READ_PARTITION
) {
523 outfile
= open(filename
,O_CREAT
|O_TRUNC
|O_WRONLY
|O_BINARY
,S_IREAD
|S_IWRITE
);
529 if (read_partition(&ipod
, outfile
) < 0) {
530 fprintf(stderr
,"[ERR] --read-partition failed.\n");
532 fprintf(stderr
,"[INFO] Partition extracted to %s.\n",filename
);
535 } else if (action
==WRITE_PARTITION
) {
536 if (ipod_reopen_rw(&ipod
) < 0) {
540 infile
= open(filename
,O_RDONLY
|O_BINARY
);
546 /* Check filesize is <= partition size */
547 inputsize
=filesize(infile
);
549 if (inputsize
<= (ipod
.pinfo
[0].size
*ipod
.sector_size
)) {
550 fprintf(stderr
,"[INFO] Input file is %u bytes\n",inputsize
);
551 if (write_partition(&ipod
,infile
) < 0) {
552 fprintf(stderr
,"[ERR] --write-partition failed.\n");
554 fprintf(stderr
,"[INFO] %s restored to partition\n",filename
);
557 fprintf(stderr
,"[ERR] File is too large for firmware partition, aborting.\n");
562 } else if (action
==FORMAT_PARTITION
) {
563 printf("WARNING!!! YOU ARE ABOUT TO USE AN EXPERIMENTAL FEATURE.\n");
564 printf("ALL DATA ON YOUR IPOD WILL BE ERASED.\n");
565 printf("Are you sure you want to format your ipod? (y/n):");
567 if (fgets(yesno
,4,stdin
)) {
569 if (ipod_reopen_rw(&ipod
) < 0) {
573 if (format_partition(&ipod
,1) < 0) {
574 fprintf(stderr
,"[ERR] Format failed.\n");
577 fprintf(stderr
,"[INFO] Format cancelled.\n");
580 } else if (action
==CONVERT_TO_FAT32
) {
582 printf("[ERR] Ipod is already FAT32, aborting\n");
584 printf("WARNING!!! YOU ARE ABOUT TO USE AN EXPERIMENTAL FEATURE.\n");
585 printf("ALL DATA ON YOUR IPOD WILL BE ERASED.\n");
586 printf("Are you sure you want to convert your ipod to FAT32? (y/n):");
588 if (fgets(yesno
,4,stdin
)) {
590 if (ipod_reopen_rw(&ipod
) < 0) {
594 if (write_dos_partition_table(&ipod
) < 0) {
595 fprintf(stderr
,"[ERR] Partition conversion failed.\n");
598 if (format_partition(&ipod
,1) < 0) {
599 fprintf(stderr
,"[ERR] Format failed.\n");
602 fprintf(stderr
,"[INFO] Format cancelled.\n");
611 if (action
==INTERACTIVE
) {
612 printf("Press ENTER to exit ipodpatcher :");
613 fgets(yesno
,4,stdin
);