1 /***************************************************************************
3 * Open \______ \ ____ ____ | | _\_ |__ _______ ___
4 * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
5 * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
6 * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
8 * $Id: ipodpatcher.c 12237 2007-02-08 21:31:38Z dave $
10 * Copyright (C) 2006-2007 Dave Chapman
12 * All files in this archive are subject to the GNU General Public License.
13 * See the file COPYING in the source tree root for full license agreement.
15 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
16 * KIND, either express or implied.
18 ****************************************************************************/
26 #include <sys/types.h>
29 #include "ipodpatcher.h"
32 #define VERSION "2.0 with v2.0 bootloaders"
56 void print_macpod_warning(void)
58 printf("[INFO] ************************************************************************\n");
59 printf("[INFO] *** WARNING FOR ROCKBOX USERS\n");
60 printf("[INFO] *** You must convert this ipod to FAT32 format (aka a \"winpod\")\n");
61 printf("[INFO] *** if you want to run Rockbox. Rockbox WILL NOT work on this ipod.\n");
62 printf("[INFO] *** See http://www.rockbox.org/twiki/bin/view/Main/IpodConversionToFAT32\n");
63 printf("[INFO] ************************************************************************\n");
66 void print_usage(void)
68 fprintf(stderr
,"Usage: ipodpatcher --scan\n");
70 fprintf(stderr
," or ipodpatcher [DISKNO] [action]\n");
72 fprintf(stderr
," or ipodpatcher [device] [action]\n");
75 fprintf(stderr
,"Where [action] is one of the following options:\n");
77 fprintf(stderr
," --install\n");
79 fprintf(stderr
," -l, --list\n");
80 fprintf(stderr
," -r, --read-partition bootpartition.bin\n");
81 fprintf(stderr
," -w, --write-partition bootpartition.bin\n");
82 fprintf(stderr
," -rf, --read-firmware filename.ipod\n");
83 fprintf(stderr
," -rfb, --read-firmware-bin filename.bin\n");
84 fprintf(stderr
," -wf, --write-firmware filename.ipod\n");
85 fprintf(stderr
," -wfb, --write-firmware-bin filename.bin\n");
87 fprintf(stderr
," -we, --write-embedded\n");
89 fprintf(stderr
," -a, --add-bootloader filename.ipod\n");
90 fprintf(stderr
," -ab, --add-bootloader-bin filename.bin\n");
91 fprintf(stderr
," -d, --delete-bootloader\n");
92 fprintf(stderr
," -f, --format\n");
93 fprintf(stderr
," -c, --convert\n");
94 fprintf(stderr
," --read-aupd filename.bin\n");
95 fprintf(stderr
," --write-aupd filename.bin\n");
99 fprintf(stderr
,"DISKNO is the number (e.g. 2) Windows has assigned to your ipod's hard disk.\n");
100 fprintf(stderr
,"The first hard disk in your computer (i.e. C:\\) will be disk 0, the next disk\n");
101 fprintf(stderr
,"will be disk 1 etc. ipodpatcher will refuse to access a disk unless it\n");
102 fprintf(stderr
,"can identify it as being an ipod.\n");
103 fprintf(stderr
,"\n");
105 #if defined(linux) || defined (__linux)
106 fprintf(stderr
,"\"device\" is the device node (e.g. /dev/sda) assigned to your ipod.\n");
107 #elif defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)
108 fprintf(stderr
,"\"device\" is the device node (e.g. /dev/da1) assigned to your ipod.\n");
109 #elif defined(__APPLE__) && defined(__MACH__)
110 fprintf(stderr
,"\"device\" is the device node (e.g. /dev/disk1) assigned to your ipod.\n");
112 fprintf(stderr
,"ipodpatcher will refuse to access a disk unless it can identify it as being\n");
113 fprintf(stderr
,"an ipod.\n");
117 void display_partinfo(struct ipod_t
* ipod
)
120 double sectors_per_MB
= (1024.0*1024.0)/ipod
->sector_size
;
122 printf("[INFO] Part Start Sector End Sector Size (MB) Type\n");
123 for ( i
= 0; i
< 4; i
++ ) {
124 if (ipod
->pinfo
[i
].start
!= 0) {
125 printf("[INFO] %d %10ld %10ld %10.1f %s (0x%02x)\n",
127 (long int)ipod
->pinfo
[i
].start
,
128 (long int)ipod
->pinfo
[i
].start
+ipod
->pinfo
[i
].size
-1,
129 ipod
->pinfo
[i
].size
/sectors_per_MB
,
130 get_parttype(ipod
->pinfo
[i
].type
),
131 (int)ipod
->pinfo
[i
].type
);
137 int main(int argc
, char* argv
[])
143 unsigned int inputsize
;
145 int action
= SHOW_INFO
;
149 fprintf(stderr
,"ipodpatcher v" VERSION
" - (C) Dave Chapman 2006-2007\n");
150 fprintf(stderr
,"This is free software; see the source for copying conditions. There is NO\n");
151 fprintf(stderr
,"warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\n");
153 if ((argc
> 1) && ((strcmp(argv
[1],"-h")==0) || (strcmp(argv
[1],"--help")==0))) {
158 if (ipod_alloc_buffer(§orbuf
,BUFFER_SIZE
) < 0) {
159 fprintf(stderr
,"Failed to allocate memory buffer\n");
162 if ((argc
> 1) && (strcmp(argv
[1],"--scan")==0)) {
163 if (ipod_scan(&ipod
) == 0)
164 fprintf(stderr
,"[ERR] No ipods found.\n");
168 /* If the first parameter doesn't start with -, then we interpret it as a device */
169 if ((argc
> 1) && (argv
[1][0] != '-')) {
172 snprintf(ipod
.diskname
,sizeof(ipod
.diskname
),"\\\\.\\PhysicalDrive%s",argv
[1]);
174 strncpy(ipod
.diskname
,argv
[1],sizeof(ipod
.diskname
));
178 /* Autoscan for ipods */
179 n
= ipod_scan(&ipod
);
181 fprintf(stderr
,"[ERR] No ipods found, aborting\n");
182 fprintf(stderr
,"[ERR] Please connect your ipod and ensure it is in disk mode\n");
183 #if defined(__APPLE__) && defined(__MACH__)
184 fprintf(stderr
,"[ERR] Also ensure that itunes is closed, and that your ipod is not mounted.\n");
185 #elif !defined(__WIN32__)
187 fprintf(stderr
,"[ERR] You may also need to run ipodpatcher as root.\n");
190 fprintf(stderr
,"[ERR] Please refer to the Rockbox manual if you continue to have problems.\n");
192 fprintf(stderr
,"[ERR] %d ipods found, aborting\n",n
);
193 fprintf(stderr
,"[ERR] Please connect only one ipod and re-run ipodpatcher.\n");
199 printf("\nPress ENTER to exit ipodpatcher :");
200 fgets(yesno
,4,stdin
);
210 action
= INTERACTIVE
;
216 if ((strcmp(argv
[i
],"-l")==0) || (strcmp(argv
[i
],"--list")==0)) {
217 action
= LIST_IMAGES
;
220 } else if (strcmp(argv
[i
],"--install")==0) {
224 } else if ((strcmp(argv
[i
],"-d")==0) ||
225 (strcmp(argv
[i
],"--delete-bootloader")==0)) {
226 action
= DELETE_BOOTLOADER
;
228 } else if ((strcmp(argv
[i
],"-a")==0) ||
229 (strcmp(argv
[i
],"--add-bootloader")==0)) {
230 action
= ADD_BOOTLOADER
;
231 type
= FILETYPE_DOT_IPOD
;
233 if (i
== argc
) { print_usage(); return 1; }
236 } else if ((strcmp(argv
[i
],"-ab")==0) ||
237 (strcmp(argv
[i
],"--add-bootloader-bin")==0)) {
238 action
= ADD_BOOTLOADER
;
239 type
= FILETYPE_DOT_BIN
;
241 if (i
== argc
) { print_usage(); return 1; }
244 } else if ((strcmp(argv
[i
],"-rf")==0) ||
245 (strcmp(argv
[i
],"--read-firmware")==0)) {
246 action
= READ_FIRMWARE
;
247 type
= FILETYPE_DOT_IPOD
;
249 if (i
== argc
) { print_usage(); return 1; }
252 } else if ((strcmp(argv
[i
],"-rfb")==0) ||
253 (strcmp(argv
[i
],"--read-firmware-bin")==0)) {
254 action
= READ_FIRMWARE
;
255 type
= FILETYPE_DOT_BIN
;
257 if (i
== argc
) { print_usage(); return 1; }
261 } else if ((strcmp(argv
[i
],"-we")==0) ||
262 (strcmp(argv
[i
],"--write-embedded")==0)) {
263 action
= WRITE_FIRMWARE
;
264 type
= FILETYPE_INTERNAL
;
265 filename
="[embedded bootloader]"; /* Only displayed for user */
268 } else if ((strcmp(argv
[i
],"-wf")==0) ||
269 (strcmp(argv
[i
],"--write-firmware")==0)) {
270 action
= WRITE_FIRMWARE
;
271 type
= FILETYPE_DOT_IPOD
;
273 if (i
== argc
) { print_usage(); return 1; }
276 } else if ((strcmp(argv
[i
],"-wfb")==0) ||
277 (strcmp(argv
[i
],"--write-firmware-bin")==0)) {
278 action
= WRITE_FIRMWARE
;
279 type
= FILETYPE_DOT_BIN
;
281 if (i
== argc
) { print_usage(); return 1; }
284 } else if ((strcmp(argv
[i
],"-r")==0) ||
285 (strcmp(argv
[i
],"--read-partition")==0)) {
286 action
= READ_PARTITION
;
288 if (i
== argc
) { print_usage(); return 1; }
291 } else if ((strcmp(argv
[i
],"-w")==0) ||
292 (strcmp(argv
[i
],"--write-partition")==0)) {
293 action
= WRITE_PARTITION
;
295 if (i
== argc
) { print_usage(); return 1; }
298 } else if ((strcmp(argv
[i
],"-v")==0) ||
299 (strcmp(argv
[i
],"--verbose")==0)) {
302 } else if ((strcmp(argv
[i
],"-f")==0) ||
303 (strcmp(argv
[i
],"--format")==0)) {
304 action
= FORMAT_PARTITION
;
306 } else if (strcmp(argv
[i
],"--read-aupd")==0) {
309 if (i
== argc
) { print_usage(); return 1; }
312 } else if (strcmp(argv
[i
],"--write-aupd")==0) {
315 if (i
== argc
) { print_usage(); return 1; }
318 } else if ((strcmp(argv
[i
],"-c")==0) ||
319 (strcmp(argv
[i
],"--convert")==0)) {
320 action
= CONVERT_TO_FAT32
;
323 print_usage(); return 1;
327 if (ipod
.diskname
[0]==0) {
332 if (ipod_open(&ipod
, 0) < 0) {
336 fprintf(stderr
,"[INFO] Reading partition table from %s\n",ipod
.diskname
);
337 fprintf(stderr
,"[INFO] Sector size is %d bytes\n",ipod
.sector_size
);
339 if (read_partinfo(&ipod
,0) < 0) {
343 display_partinfo(&ipod
);
345 if (ipod
.pinfo
[0].start
==0) {
346 fprintf(stderr
,"[ERR] No partition 0 on disk:\n");
347 display_partinfo(&ipod
);
351 read_directory(&ipod
);
353 if (ipod
.nimages
<= 0) {
354 fprintf(stderr
,"[ERR] Failed to read firmware directory - nimages=%d\n",ipod
.nimages
);
358 if (getmodel(&ipod
,(ipod
.ipod_directory
[0].vers
>>8)) < 0) {
359 fprintf(stderr
,"[ERR] Unknown version number in firmware (%08x)\n",
360 ipod
.ipod_directory
[0].vers
);
364 printf("[INFO] Ipod model: %s (\"%s\")\n",ipod
.modelstr
,
365 ipod
.macpod
? "macpod" : "winpod");
367 if (ipod
.ipod_directory
[0].vers
== 0x10000) {
368 fprintf(stderr
,"[ERR] *** ipodpatcher does not support the 2nd Generation Nano! ***\n");
370 printf("Press ENTER to exit ipodpatcher :");
371 fgets(yesno
,4,stdin
);
377 print_macpod_warning();
380 if (action
==LIST_IMAGES
) {
383 } else if (action
==INTERACTIVE
) {
385 printf("Enter i to install the Rockbox bootloader, u to uninstall\n or c to cancel and do nothing (i/u/c) :");
387 if (fgets(yesno
,4,stdin
)) {
389 if (ipod_reopen_rw(&ipod
) < 0) {
393 if (add_bootloader(&ipod
, NULL
, FILETYPE_INTERNAL
)==0) {
394 fprintf(stderr
,"[INFO] Bootloader installed successfully.\n");
396 fprintf(stderr
,"[ERR] --install failed.\n");
398 } else if (yesno
[0]=='u') {
399 if (ipod_reopen_rw(&ipod
) < 0) {
403 if (delete_bootloader(&ipod
)==0) {
404 fprintf(stderr
,"[INFO] Bootloader removed.\n");
406 fprintf(stderr
,"[ERR] Bootloader removal failed.\n");
411 } else if (action
==DELETE_BOOTLOADER
) {
412 if (ipod_reopen_rw(&ipod
) < 0) {
416 if (ipod
.ipod_directory
[0].entryOffset
==0) {
417 fprintf(stderr
,"[ERR] No bootloader detected.\n");
419 if (delete_bootloader(&ipod
)==0) {
420 fprintf(stderr
,"[INFO] Bootloader removed.\n");
422 fprintf(stderr
,"[ERR] --delete-bootloader failed.\n");
425 } else if (action
==ADD_BOOTLOADER
) {
426 if (ipod_reopen_rw(&ipod
) < 0) {
430 if (add_bootloader(&ipod
, filename
, type
)==0) {
431 fprintf(stderr
,"[INFO] Bootloader %s written to device.\n",filename
);
433 fprintf(stderr
,"[ERR] --add-bootloader failed.\n");
436 } else if (action
==INSTALL
) {
437 if (ipod_reopen_rw(&ipod
) < 0) {
441 if (add_bootloader(&ipod
, NULL
, FILETYPE_INTERNAL
)==0) {
442 fprintf(stderr
,"[INFO] Bootloader installed successfully.\n");
444 fprintf(stderr
,"[ERR] --install failed.\n");
447 } else if (action
==WRITE_FIRMWARE
) {
448 if (ipod_reopen_rw(&ipod
) < 0) {
452 if (write_firmware(&ipod
, filename
,type
)==0) {
453 fprintf(stderr
,"[INFO] Firmware %s written to device.\n",filename
);
455 fprintf(stderr
,"[ERR] --write-firmware failed.\n");
457 } else if (action
==READ_FIRMWARE
) {
458 if (read_firmware(&ipod
, filename
, type
)==0) {
459 fprintf(stderr
,"[INFO] Firmware read to file %s.\n",filename
);
461 fprintf(stderr
,"[ERR] --read-firmware failed.\n");
463 } else if (action
==READ_AUPD
) {
464 if (read_aupd(&ipod
, filename
)==0) {
465 fprintf(stderr
,"[INFO] AUPD image read to file %s.\n",filename
);
467 fprintf(stderr
,"[ERR] --read-aupd failed.\n");
469 } else if (action
==WRITE_AUPD
) {
470 if (ipod_reopen_rw(&ipod
) < 0) {
474 if (write_aupd(&ipod
, filename
)==0) {
475 fprintf(stderr
,"[INFO] AUPD image %s written to device.\n",filename
);
477 fprintf(stderr
,"[ERR] --write-aupd failed.\n");
479 } else if (action
==READ_PARTITION
) {
480 outfile
= open(filename
,O_CREAT
|O_TRUNC
|O_WRONLY
|O_BINARY
,S_IREAD
|S_IWRITE
);
486 if (read_partition(&ipod
, outfile
) < 0) {
487 fprintf(stderr
,"[ERR] --read-partition failed.\n");
489 fprintf(stderr
,"[INFO] Partition extracted to %s.\n",filename
);
492 } else if (action
==WRITE_PARTITION
) {
493 if (ipod_reopen_rw(&ipod
) < 0) {
497 infile
= open(filename
,O_RDONLY
|O_BINARY
);
503 /* Check filesize is <= partition size */
504 inputsize
=filesize(infile
);
506 if (inputsize
<= (ipod
.pinfo
[0].size
*ipod
.sector_size
)) {
507 fprintf(stderr
,"[INFO] Input file is %u bytes\n",inputsize
);
508 if (write_partition(&ipod
,infile
) < 0) {
509 fprintf(stderr
,"[ERR] --write-partition failed.\n");
511 fprintf(stderr
,"[INFO] %s restored to partition\n",filename
);
514 fprintf(stderr
,"[ERR] File is too large for firmware partition, aborting.\n");
519 } else if (action
==FORMAT_PARTITION
) {
520 printf("WARNING!!! YOU ARE ABOUT TO USE AN EXPERIMENTAL FEATURE.\n");
521 printf("ALL DATA ON YOUR IPOD WILL BE ERASED.\n");
522 printf("Are you sure you want to format your ipod? (y/n):");
524 if (fgets(yesno
,4,stdin
)) {
526 if (ipod_reopen_rw(&ipod
) < 0) {
530 if (format_partition(&ipod
,1) < 0) {
531 fprintf(stderr
,"[ERR] Format failed.\n");
534 fprintf(stderr
,"[INFO] Format cancelled.\n");
537 } else if (action
==CONVERT_TO_FAT32
) {
539 printf("[ERR] Ipod is already FAT32, aborting\n");
541 printf("WARNING!!! YOU ARE ABOUT TO USE AN EXPERIMENTAL FEATURE.\n");
542 printf("ALL DATA ON YOUR IPOD WILL BE ERASED.\n");
543 printf("Are you sure you want to convert your ipod to FAT32? (y/n):");
545 if (fgets(yesno
,4,stdin
)) {
547 if (ipod_reopen_rw(&ipod
) < 0) {
551 if (write_dos_partition_table(&ipod
) < 0) {
552 fprintf(stderr
,"[ERR] Partition conversion failed.\n");
555 if (format_partition(&ipod
,1) < 0) {
556 fprintf(stderr
,"[ERR] Format failed.\n");
559 fprintf(stderr
,"[INFO] Format cancelled.\n");
568 if (action
==INTERACTIVE
) {
569 printf("Press ENTER to exit ipodpatcher :");
570 fgets(yesno
,4,stdin
);