isohybrid: allow selecting the _c and _f versions of the prefix
[people/sha0/syslinux.git] / utils / isohybrid.in
1 #!/usr/bin/perl
2 ## -----------------------------------------------------------------------
3 ##
4 ##   Copyright 2002-2008 H. Peter Anvin - All Rights Reserved
5 ##   Copyright 2009 Intel Corporation; author: H. Peter Anvin
6 ##
7 ##   This program is free software; you can redistribute it and/or modify
8 ##   it under the terms of the GNU General Public License as published by
9 ##   the Free Software Foundation, Inc., 53 Temple Place Ste 330,
10 ##   Boston MA 02111-1307, USA; either version 2 of the License, or
11 ##   (at your option) any later version; incorporated herein by reference.
12 ##
13 ## -----------------------------------------------------------------------
14
15 #
16 # Post-process an ISO 9660 image generated with mkisofs/genisoimage
17 # to allow "hybrid booting" as a CD-ROM or as a hard disk.
18 #
19
20 use bytes;
21 use Fcntl;
22
23 # User-specifyable options
24 %opt = (
25     # Fake geometry (zipdrive-style...)
26     'h'      => 64,
27     's'      => 32,
28     # Partition number
29     'entry'  => 1,
30     # Partition offset
31     'offset' => 0,
32     # Partition type
33     'type'   => 0x17,           # "Windows hidden IFS"
34     # MBR ID
35     'id'     => undef,
36 );
37
38 %valid_range = (
39     'h'      => [1, 256],
40     's'      => [1, 63],
41     'entry'  => [1, 4],
42     'offset' => [0, 64],
43     'type'   => [0, 255],
44     'id'     => [0, 0xffffffff],
45     'hd0'    => [0, 2],
46 );
47
48 # Boolean options just set other options
49 %bool_opt = (
50     'nohd0'    => ['hd0', 0],
51     'forcehd0' => ['hd0', 1],
52     'ctrlhd0'  => ['hd0', 2],
53 );
54
55 sub usage() {
56     print STDERR "Usage: $0 [options] filename.iso\n",
57     "Options:\n",
58     "  -h          Number of default geometry heads\n",
59     "  -s          Number of default geometry sectors\n",
60     "  -entry      Specify partition entry number (1-4)\n",
61     "  -offset     Specify partition offset (default 0)\n",
62     "  -type       Specify partition type (default 0x17)\n",
63     "  -id         Specify MBR ID (default random)\n",
64     "  -forcehd0   Always assume we are loaded as disk ID 0\n",
65     "  -ctrlhd0    Assume disk ID 0 if the Ctrl key is pressed\n",
66     exit 1;
67 }
68
69 # Parse a C-style integer (decimal/octal/hex)
70 sub doh($) {
71     my($n) = @_;
72     return ($n =~ /^0/) ? oct $n : $n+0;
73 }
74
75 sub get_random() {
76     # Get a 32-bit random number
77     my $rfd, $rnd;
78     my $rid;
79
80     if (open($rfd, "< /dev/urandom\0") && read($rfd, $rnd, 4) == 4) {
81         $rid = unpack("V", $rnd);
82     }
83
84     close($rfd) if (defined($rfd));
85     return $rid if (defined($rid));
86
87     # This sucks but is better than nothing...
88     return ($$+time()) & 0xffffffff;
89 }
90
91 sub get_hex_data() {
92     my $mbr = '';
93     my $line, $byte;
94     while ( $line = <DATA> ) {
95         chomp $line;
96         last if ($line eq '*');
97         foreach $byte ( split(/\s+/, $line) ) {
98             $mbr .= chr(hex($byte));
99         }
100     }
101     return $mbr;
102 }
103
104 while ($ARGV[0] =~ /^\-(.*)$/) {
105     $o = $1;
106     shift @ARGV;
107     if (defined($bool_opt{$o})) {
108         ($o, $v) = @{$bool_opt{$o}};
109         $opt{$o} = $v;
110     } elsif (exists($opt{$o})) {
111         $opt{$o} = doh(shift @ARGV);
112         if (defined($valid_range{$o})) {
113             ($l, $h) = @{$valid_range{$o}};
114             if ($opt{$o} < $l || $opt{$o} > $h) {
115                 die "$0: valid values for the -$o parameter are $l to $h\n";
116             }
117         }
118     } else {
119         usage();
120     }
121 }
122
123 ($file) = @ARGV;
124
125 if (!defined($file)) {
126     usage();
127 }
128
129 open(FILE, "+< $file\0") or die "$0: cannot open $file: $!\n";
130 binmode FILE;
131
132 #
133 # First, actually figure out where mkisofs hid isolinux.bin
134 #
135 seek(FILE, 17*2048, SEEK_SET) or die "$0: $file: $!\n";
136 read(FILE, $boot_record, 2048) == 2048 or die "$0: $file: read error\n";
137 ($br_sign, $br_cat_offset) = unpack("a71V", $boot_record);
138 if ($br_sign ne ("\0CD001\1EL TORITO SPECIFICATION" . ("\0" x 41))) {
139     die "$0: $file: no boot record found\n";
140 }
141 seek(FILE, $br_cat_offset*2048, SEEK_SET) or die "$0: $file: $!\n";
142 read(FILE, $boot_cat, 2048) == 2048 or die "$0: $file: read error\n";
143
144 # We must have a Validation Entry followed by a Default Entry...
145 # no fanciness allowed for the Hybrid mode [XXX: might relax this later]
146 @ve = unpack("v16", $boot_cat);
147 $cs = 0;
148 for ($i = 0; $i < 16; $i++) {
149     $cs += $ve[$i];
150 }
151 if ($ve[0] != 0x0001 || $ve[15] != 0xaa55 || $cs & 0xffff) {
152     die "$0: $file: invalid boot catalog\n";
153 }
154 ($de_boot, $de_media, $de_seg, $de_sys, $de_mbz1, $de_count, 
155  $de_lba, $de_mbz2) = unpack("CCvCCvVv", substr($boot_cat, 32, 32));
156 if ($de_boot != 0x88 || $de_media != 0 ||
157     ($de_segment != 0 && $de_segment != 0x7c0) || $de_count != 4) {
158     die "$0: $file: unexpected boot catalog parameters\n";
159 }
160
161 # Now $de_lba should contain the CD sector number for isolinux.bin
162 seek(FILE, $de_lba*2048+0x40, SEEK_SET) or die "$0: $file: $!\n";
163 read(FILE, $ibsig, 4);
164 if ($ibsig ne "\xfb\xc0\x78\x70") {
165     die "$0: $file: bootloader is missing isolinux.bin hybrid signature\n".
166         "Note: isolinux-debug.bin does not support hybrid booting\n";
167 }
168
169 # Get the total size of the image
170 (@imgstat = stat(FILE)) or die "$0: $file: $!\n";
171 $imgsize = $imgstat[7];
172 if (!$imgsize) {
173     die "$0: $file: cannot determine length of file\n";
174 }
175 # Target image size: round up to a multiple of $h*$s*512
176 $h = $opt{'h'};
177 $s = $opt{'s'};
178 $cylsize = $h*$s*512;
179 $frac = $imgsize % $cylsize;
180 $padding = ($frac > 0) ? $cylsize - $frac : 0;
181 $imgsize += $padding;
182 $c = int($imgsize/$cylsize);
183 if ($c > 1024) {
184     print STDERR "Warning: more than 1024 cylinders ($c).\n";
185     print STDERR "Not all BIOSes will be able to boot this device.\n";
186     $cc = 1024;
187 } else {
188     $cc = $c;
189 }
190
191 # Preserve id when run again
192 if (defined($opt{'id'})) {
193     $id = $opt{'id'};
194 } else {
195     seek(FILE, 440, SEEK_SET) or die "$0: $file: $!\n";
196     read(FILE, $id, 4);
197     if ($id eq "\x00\x00\x00\x00") {
198         $id = pack("V", get_random());
199     }
200 }
201
202 # Print the MBR and partition table
203 seek(FILE, 0, SEEK_SET) or die "$0: $file: $!\n";
204
205 for ($i = 0; $i <= $opt{'hd0'}; $i++) {
206     $mbr = get_hex_data();
207 }
208 if ( length($mbr) > 432 ) {
209     die "$0: Bad MBR code\n";
210 }
211
212 $mbr .= "\0" x (432 - length($mbr));
213
214 $mbr .= pack("VV", $de_lba*4, 0);       # Offset 432: LBA of isolinux.bin
215 $mbr .= $id;                            # Offset 440: MBR ID
216 $mbr .= "\0\0";                         # Offset 446: actual partition table
217
218 # Print partition table
219 $offset  = $opt{'offset'};
220 $psize   = $c*$h*$s;
221 $bhead   = int($offset/$s) % $h;
222 $bsect   = ($offset % $s) + 1;
223 $bcyl    = int($offset/($h*$s));
224 $bsect  += ($bcyl & 0x300) >> 2;
225 $bcyl   &= 0xff;
226 $ehead   = $h-1;
227 $esect   = $s + ((($cc-1) & 0x300) >> 2);
228 $ecyl    = ($cc-1) & 0xff;
229 $fstype  = $opt{'type'};        # Partition type
230 $pentry  = $opt{'entry'};       # Partition slot
231
232 for ( $i = 1 ; $i <= 4 ; $i++ ) {
233     if ( $i == $pentry ) {
234         $mbr .= pack("CCCCCCCCVV", 0x80, $bhead, $bsect, $bcyl, $fstype,
235                      $ehead, $esect, $ecyl, 0, $psize);
236     } else {
237         $mbr .= "\0" x 16;
238     }
239 }
240 $mbr .= "\x55\xaa";
241
242 print FILE $mbr;
243
244 # Pad the image to a fake cylinder boundary
245 seek(FILE, $imgstat[7], SEEK_SET) or die "$0: $file: $!\n";
246 if ($padding) {
247     print FILE "\0" x $padding;
248 }
249
250 # Done...
251 close(FILE);
252
253 exit 0;
254 __END__