We have a 3.9 release, update builds.pm
[maemo-rb.git] / rbutil / ipodpatcher / main.c
blob2d4ad3a796d3e5e98349c10561fb0e17b3f23191
1 /***************************************************************************
2 * __________ __ ___.
3 * Open \______ \ ____ ____ | | _\_ |__ _______ ___
4 * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
5 * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
6 * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
7 * \/ \/ \/ \/ \/
8 * $Id$
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 ****************************************************************************/
22 #include <stdio.h>
23 #include <unistd.h>
24 #include <fcntl.h>
25 #include <string.h>
26 #include <stdlib.h>
27 #include <inttypes.h>
28 #include <sys/types.h>
29 #include <sys/stat.h>
31 #include "ipodpatcher.h"
32 #include "ipodio.h"
34 #ifdef RELEASE
35 #undef VERSION
36 #define VERSION "5.0 with v4.0 bootloaders (v1.0 for 2nd Gen Nano)"
37 #endif
40 enum {
41 NONE,
42 #ifdef WITH_BOOTOBJS
43 INSTALL,
44 #endif
45 INTERACTIVE,
46 SHOW_INFO,
47 LIST_IMAGES,
48 DELETE_BOOTLOADER,
49 ADD_BOOTLOADER,
50 READ_FIRMWARE,
51 WRITE_FIRMWARE,
52 READ_AUPD,
53 WRITE_AUPD,
54 READ_PARTITION,
55 WRITE_PARTITION,
56 FORMAT_PARTITION,
57 DUMP_XML,
58 CONVERT_TO_FAT32
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");
74 #ifdef __WIN32__
75 fprintf(stderr," or ipodpatcher [DISKNO] [action]\n");
76 #else
77 fprintf(stderr," or ipodpatcher [device] [action]\n");
78 #endif
79 fprintf(stderr,"\n");
80 fprintf(stderr,"Where [action] is one of the following options:\n");
81 #ifdef WITH_BOOTOBJS
82 fprintf(stderr," --install\n");
83 #endif
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");
91 #ifdef WITH_BOOTOBJS
92 fprintf(stderr," -we, --write-embedded\n");
93 #endif
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");
106 #ifdef __WIN32__
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");
112 #else
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");
119 #endif
120 fprintf(stderr,"ipodpatcher will refuse to access a disk unless it can identify it as being\n");
121 fprintf(stderr,"an ipod.\n");
122 #endif
125 void display_partinfo(struct ipod_t* ipod)
127 int i;
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[])
147 char yesno[4];
148 int i;
149 int n;
150 int infile, outfile;
151 unsigned int inputsize;
152 char* filename;
153 int action = SHOW_INFO;
154 int type;
155 struct ipod_t ipod;
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))) {
163 print_usage();
164 return 1;
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");
174 return 0;
177 /* If the first parameter doesn't start with -, then we interpret it as a device */
178 if ((argc > 1) && (argv[1][0] != '-')) {
179 ipod.diskname[0]=0;
180 #ifdef __WIN32__
181 snprintf(ipod.diskname,sizeof(ipod.diskname),"\\\\.\\PhysicalDrive%s",argv[1]);
182 #else
183 strncpy(ipod.diskname,argv[1],sizeof(ipod.diskname));
184 #endif
185 i = 2;
186 } else {
187 /* Autoscan for ipods */
188 n = ipod_scan(&ipod);
189 if (n==0) {
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__)
195 if (geteuid()!=0) {
196 fprintf(stderr,"[ERR] You may also need to run ipodpatcher as root.\n");
198 #endif
199 fprintf(stderr,"[ERR] Please refer to the Rockbox manual if you continue to have problems.\n");
200 } else if (n > 1) {
201 fprintf(stderr,"[ERR] %d ipods found, aborting\n",n);
202 fprintf(stderr,"[ERR] Please connect only one ipod and re-run ipodpatcher.\n");
205 if (n != 1) {
206 #ifdef WITH_BOOTOBJS
207 if (argc==1) {
208 printf("\nPress ENTER to exit ipodpatcher :");
209 fgets(yesno,4,stdin);
211 #endif
212 return 0;
215 i = 1;
218 #ifdef WITH_BOOTOBJS
219 action = INTERACTIVE;
220 #else
221 action = NONE;
222 #endif
224 while (i < argc) {
225 if ((strcmp(argv[i],"-l")==0) || (strcmp(argv[i],"--list")==0)) {
226 action = LIST_IMAGES;
227 i++;
228 #ifdef WITH_BOOTOBJS
229 } else if (strcmp(argv[i],"--install")==0) {
230 action = INSTALL;
231 i++;
232 #endif
233 } else if ((strcmp(argv[i],"-d")==0) ||
234 (strcmp(argv[i],"--delete-bootloader")==0)) {
235 action = DELETE_BOOTLOADER;
236 i++;
237 } else if ((strcmp(argv[i],"-a")==0) ||
238 (strcmp(argv[i],"--add-bootloader")==0)) {
239 action = ADD_BOOTLOADER;
240 type = FILETYPE_DOT_IPOD;
241 i++;
242 if (i == argc) { print_usage(); return 1; }
243 filename=argv[i];
244 i++;
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;
249 i++;
250 if (i == argc) { print_usage(); return 1; }
251 filename=argv[i];
252 i++;
253 } else if ((strcmp(argv[i],"-rf")==0) ||
254 (strcmp(argv[i],"--read-firmware")==0)) {
255 action = READ_FIRMWARE;
256 type = FILETYPE_DOT_IPOD;
257 i++;
258 if (i == argc) { print_usage(); return 1; }
259 filename=argv[i];
260 i++;
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;
265 i++;
266 if (i == argc) { print_usage(); return 1; }
267 filename=argv[i];
268 i++;
269 #ifdef WITH_BOOTOBJS
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 */
275 i++;
276 #endif
277 } else if ((strcmp(argv[i],"-wf")==0) ||
278 (strcmp(argv[i],"--write-firmware")==0)) {
279 action = WRITE_FIRMWARE;
280 type = FILETYPE_DOT_IPOD;
281 i++;
282 if (i == argc) { print_usage(); return 1; }
283 filename=argv[i];
284 i++;
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;
289 i++;
290 if (i == argc) { print_usage(); return 1; }
291 filename=argv[i];
292 i++;
293 } else if ((strcmp(argv[i],"-r")==0) ||
294 (strcmp(argv[i],"--read-partition")==0)) {
295 action = READ_PARTITION;
296 i++;
297 if (i == argc) { print_usage(); return 1; }
298 filename=argv[i];
299 i++;
300 } else if ((strcmp(argv[i],"-w")==0) ||
301 (strcmp(argv[i],"--write-partition")==0)) {
302 action = WRITE_PARTITION;
303 i++;
304 if (i == argc) { print_usage(); return 1; }
305 filename=argv[i];
306 i++;
307 } else if ((strcmp(argv[i],"-v")==0) ||
308 (strcmp(argv[i],"--verbose")==0)) {
309 ipod_verbose++;
310 i++;
311 } else if ((strcmp(argv[i],"-f")==0) ||
312 (strcmp(argv[i],"--format")==0)) {
313 action = FORMAT_PARTITION;
314 i++;
315 } else if (strcmp(argv[i],"--read-aupd")==0) {
316 action = READ_AUPD;
317 i++;
318 if (i == argc) { print_usage(); return 1; }
319 filename=argv[i];
320 i++;
321 } else if (strcmp(argv[i],"--write-aupd")==0) {
322 action = WRITE_AUPD;
323 i++;
324 if (i == argc) { print_usage(); return 1; }
325 filename=argv[i];
326 i++;
327 } else if ((strcmp(argv[i],"-x")==0) ||
328 (strcmp(argv[i],"--dump-xml")==0)) {
329 action = DUMP_XML;
330 i++;
331 if (i == argc) { print_usage(); return 1; }
332 filename=argv[i];
333 i++;
334 } else if ((strcmp(argv[i],"-c")==0) ||
335 (strcmp(argv[i],"--convert")==0)) {
336 action = CONVERT_TO_FAT32;
337 i++;
338 } else {
339 print_usage(); return 1;
343 if (ipod.diskname[0]==0) {
344 print_usage();
345 return 1;
348 if (ipod_open(&ipod, 0) < 0) {
349 return 1;
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) {
356 return 2;
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);
364 return 3;
367 read_directory(&ipod);
369 if (ipod.nimages <= 0) {
370 fprintf(stderr,"[ERR] Failed to read firmware directory - nimages=%d\n",ipod.nimages);
371 return 1;
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);
377 return -1;
380 #ifdef __WIN32__
381 /* Windows requires the ipod in R/W mode for SCSI Inquiry */
382 if (ipod_reopen_rw(&ipod) < 0) {
383 return 5;
385 #endif
388 /* Read the XML info, and if successful, look for the ramsize
389 (only available for some models - set to 0 if not known) */
391 ipod.ramsize = 0;
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");
401 if (ipod.macpod) {
402 print_macpod_warning();
405 if (action==LIST_IMAGES) {
406 list_images(&ipod);
407 #ifdef WITH_BOOTOBJS
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)) {
413 if (yesno[0]=='i') {
414 if (ipod_reopen_rw(&ipod) < 0) {
415 return 5;
418 if (add_bootloader(&ipod, NULL, FILETYPE_INTERNAL)==0) {
419 fprintf(stderr,"[INFO] Bootloader installed successfully.\n");
420 } else {
421 fprintf(stderr,"[ERR] --install failed.\n");
423 } else if (yesno[0]=='u') {
424 if (ipod_reopen_rw(&ipod) < 0) {
425 return 5;
428 if (delete_bootloader(&ipod)==0) {
429 fprintf(stderr,"[INFO] Bootloader removed.\n");
430 } else {
431 fprintf(stderr,"[ERR] Bootloader removal failed.\n");
435 #endif
436 } else if (action==DELETE_BOOTLOADER) {
437 if (ipod_reopen_rw(&ipod) < 0) {
438 return 5;
441 if (ipod.ipod_directory[0].entryOffset==0) {
442 fprintf(stderr,"[ERR] No bootloader detected.\n");
443 } else {
444 if (delete_bootloader(&ipod)==0) {
445 fprintf(stderr,"[INFO] Bootloader removed.\n");
446 } else {
447 fprintf(stderr,"[ERR] --delete-bootloader failed.\n");
450 } else if (action==ADD_BOOTLOADER) {
451 if (ipod_reopen_rw(&ipod) < 0) {
452 return 5;
455 if (add_bootloader(&ipod, filename, type)==0) {
456 fprintf(stderr,"[INFO] Bootloader %s written to device.\n",filename);
457 } else {
458 fprintf(stderr,"[ERR] --add-bootloader failed.\n");
460 #ifdef WITH_BOOTOBJS
461 } else if (action==INSTALL) {
462 if (ipod_reopen_rw(&ipod) < 0) {
463 return 5;
466 if (add_bootloader(&ipod, NULL, FILETYPE_INTERNAL)==0) {
467 fprintf(stderr,"[INFO] Bootloader installed successfully.\n");
468 } else {
469 fprintf(stderr,"[ERR] --install failed.\n");
471 #endif
472 } else if (action==WRITE_FIRMWARE) {
473 if (ipod_reopen_rw(&ipod) < 0) {
474 return 5;
477 if (write_firmware(&ipod, filename,type)==0) {
478 fprintf(stderr,"[INFO] Firmware %s written to device.\n",filename);
479 } else {
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);
485 } else {
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);
491 } else {
492 fprintf(stderr,"[ERR] --read-aupd failed.\n");
494 } else if (action==WRITE_AUPD) {
495 if (ipod_reopen_rw(&ipod) < 0) {
496 return 5;
499 if (write_aupd(&ipod, filename)==0) {
500 fprintf(stderr,"[INFO] AUPD image %s written to device.\n",filename);
501 } else {
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");
507 return 1;
510 outfile = open(filename,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,S_IREAD|S_IWRITE);
511 if (outfile < 0) {
512 perror(filename);
513 return 4;
516 if (write(outfile, ipod.xmlinfo, ipod.xmlinfo_len) < 0) {
517 fprintf(stderr,"[ERR] --dump-xml failed.\n");
518 } else {
519 fprintf(stderr,"[INFO] XML info written to %s.\n",filename);
521 close(outfile);
522 } else if (action==READ_PARTITION) {
523 outfile = open(filename,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,S_IREAD|S_IWRITE);
524 if (outfile < 0) {
525 perror(filename);
526 return 4;
529 if (read_partition(&ipod, outfile) < 0) {
530 fprintf(stderr,"[ERR] --read-partition failed.\n");
531 } else {
532 fprintf(stderr,"[INFO] Partition extracted to %s.\n",filename);
534 close(outfile);
535 } else if (action==WRITE_PARTITION) {
536 if (ipod_reopen_rw(&ipod) < 0) {
537 return 5;
540 infile = open(filename,O_RDONLY|O_BINARY);
541 if (infile < 0) {
542 perror(filename);
543 return 2;
546 /* Check filesize is <= partition size */
547 inputsize=filesize(infile);
548 if (inputsize > 0) {
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");
553 } else {
554 fprintf(stderr,"[INFO] %s restored to partition\n",filename);
556 } else {
557 fprintf(stderr,"[ERR] File is too large for firmware partition, aborting.\n");
561 close(infile);
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)) {
568 if (yesno[0]=='y') {
569 if (ipod_reopen_rw(&ipod) < 0) {
570 return 5;
573 if (format_partition(&ipod,1) < 0) {
574 fprintf(stderr,"[ERR] Format failed.\n");
576 } else {
577 fprintf(stderr,"[INFO] Format cancelled.\n");
580 } else if (action==CONVERT_TO_FAT32) {
581 if (!ipod.macpod) {
582 printf("[ERR] Ipod is already FAT32, aborting\n");
583 } else {
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)) {
589 if (yesno[0]=='y') {
590 if (ipod_reopen_rw(&ipod) < 0) {
591 return 5;
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");
601 } else {
602 fprintf(stderr,"[INFO] Format cancelled.\n");
608 ipod_close(&ipod);
610 #ifdef WITH_BOOTOBJS
611 if (action==INTERACTIVE) {
612 printf("Press ENTER to exit ipodpatcher :");
613 fgets(yesno,4,stdin);
615 #endif
617 return 0;