#!/usr/bin/perl
#
# Guillaume Cottenceau (gc@mandrakesoft.com)
#
# Copyright 2002 Mandrakesoft
#
# This software may be freely redistributed under the terms of the GNU
# public license.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#

sub arch() {
    my $t = `uname -m`;
    chomp $t;
    $t;
}
sub if_($@) {
    my $b = shift;
    $b or return ();
    wantarray || @_ <= 1 or die("if_ called in scalar context with more than one argument " . join(":", caller()));
    wantarray ? @_ : $_[0];
}
sub cat_ { local *F; open F, $_[0] or $_[1] ? die "cat of file $_[0] failed: $!\n" : return; my @l = <F>; wantarray ? @l : join '', @l }
sub member { my $e = shift; foreach (@_) { $e eq $_ and return 1 } 0 }

@known_boot_loaders = qw(lilo grub yaboot);

# keep this in sync with DrakX
@MBR_signatures = (
    [ 'empty', 0, "\0\0\0\0" ],
    [ 'grub', 0x6,  "GRUB" ],
    [ 'grub', 0, "\xEBG", 0x17d, "stage1 \0" ],
    [ 'grub', 0, "\xEBH", 0x17e, "stage1 \0" ],
    [ 'grub', 0, "\xEBH", 0x18a, "stage1 \0" ],
    [ 'grub', 0, "\xEBH", 0x181, "GRUB \0" ],
    [ 'lilo', 0x2,  "LILO" ],
    [ 'lilo', 0x6,  "LILO" ],
if_(arch() =~ /ppc/,
    map { [ 'yaboot', 0, "PM", 0x200 * $_ + 0x10, "bootstrap\0" ] } 0 .. 61
),
);

sub typeFromMagic {
    my $f = shift;
    local *F; sysopen F, $f, 0 or return;

    my $tmp;
  M: foreach (@MBR_signatures) {
	my ($name, @l) = @$_;
	while (@l) {
	    my ($offset, $signature) = splice(@l, 0, 2);
	    sysseek(F, $offset, 0) or next M;
	    sysread(F, $tmp, length $signature);
	    $tmp eq $signature or next M;
	}
	return $name;
    }
    return 0;
}


my (undef, undef, @parts) = cat_('/proc/partitions');

my @possibilities;

P: foreach (@parts) {
    my (undef, undef, $blocks, $dev) = split or next;
    next if $blocks <= 1;
    my $type = typeFromMagic("/dev/$dev");
    $type && member($type, @known_boot_loaders) and push @possibilities, [ $dev, $type ];
}

my $choice;

if (!@possibilities) {
    print "No known Linux bootloader has been found, nothing to do.\n";
} elsif (@possibilities == 1) {
    print "I've found a Linux bootloader only on </dev/$possibilities[0]->[0]>.\n\n";
    $choice = $possibilities[0];
} else {
    print "I've found the following Linux bootloaders:\n";
    my $i;
    print "\t", ++$i, ": <$_->[1]> \ton <$_->[0]>\n" foreach @possibilities;
    print "\n";
    print "Which disk/partition do you want to overwrite with the Windows bootloader?\n";
    print "\t<enter the number or press 'n' and Enter to cancel> ";
    <STDIN> !~ /^(\d+)$/i && $1 >= 1 and $choice = $possibilities[$1-1];
}

if ($choice) {
    print "I'm going to overwrite bootloader on </dev/$choice->[0]> with
Windows bootloader.

Ok? <press Enter to continue, 'n' and Enter to cancel> ";
    <STDIN> =~ /^n/i and exit 0;
    
    system("/bin/dd if=/usr/lib/extipl/aldebaran.bin of=/dev/$choice->[0]\n") and print "\tFailed!\n";
}

#-------------------------------------------------
#- $Log$
#- Revision 1.2  2004/07/20 02:42:12  prigaux
#- MandrakeSoft -> Mandrakesoft
#-
#- Revision 1.1  2002/02/27 13:31:30  gc
#- add "restore Windows Boot Loader" to rescue
#-
#-
on value='author'>author</option>
<option value='committer'>committer</option>
<option value='range'>range</option>
</select>
<input class='txt' type='search' size='10' name='q' value=''/>
<input type='submit' value='search'/>
</form>
</td></tr></table>
<div class='path'>path: <a href='/software/ldetect-lst/commit/?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>root</a>/<a href='/software/ldetect-lst/commit/lst?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>lst</a>/<a href='/software/ldetect-lst/commit/lst/pcitable?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>pcitable</a></div><div class='content'><div class='cgit-panel'><b>diff options</b><form method='get'><input type='hidden' name='h' value='V0_1_9_18mdk'/><input type='hidden' name='id' value='45fde6767af82053307942600680f19bd848a27d'/><table><tr><td colspan='2'/></tr><tr><td class='label'>context:</td><td class='ctrl'><select name='context' onchange='this.form.submit();'><option value='1'>1</option><option value='2'>2</option><option value='3' selected='selected'>3</option><option value='4'>4</option><option value='5'>5</option><option value='6'>6</option><option value='7'>7</option><option value='8'>8</option><option value='9'>9</option><option value='10'>10</option><option value='15'>15</option><option value='20'>20</option><option value='25'>25</option><option value='30'>30</option><option value='35'>35</option><option value='40'>40</option></select></td></tr><tr><td class='label'>space:</td><td class='ctrl'><select name='ignorews' onchange='this.form.submit();'><option value='0' selected='selected'>include</option><option value='1'>ignore</option></select></td></tr><tr><td class='label'>mode:</td><td class='ctrl'><select name='dt' onchange='this.form.submit();'><option value='0' selected='selected'>unified</option><option value='1'>ssdiff</option><option value='2'>stat only</option></select></td></tr><tr><td/><td class='ctrl'><noscript><input type='submit' value='reload'/></noscript></td></tr></table></form></div><table summary='commit info' class='commit-info'>
<tr><th>author</th><td>Thierry Vignaud &lt;tvignaud@mandriva.org&gt;</td><td class='right'>2004-02-08 00:38:00 +0000</td></tr>
<tr><th>committer</th><td>Thierry Vignaud &lt;tvignaud@mandriva.org&gt;</td><td class='right'>2004-02-08 00:38:00 +0000</td></tr>
<tr><th>commit</th><td colspan='2' class='sha1'><a href='/software/ldetect-lst/commit/lst/pcitable?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>45fde6767af82053307942600680f19bd848a27d</a> (<a href='/software/ldetect-lst/patch/lst/pcitable?id=45fde6767af82053307942600680f19bd848a27d'>patch</a>)</td></tr>
<tr><th>tree</th><td colspan='2' class='sha1'><a href='/software/ldetect-lst/tree/?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>0ac894557f211c63c73a8003f45008312410a561</a> /<a href='/software/ldetect-lst/tree/lst/pcitable?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>lst/pcitable</a></td></tr>
<tr><th>parent</th><td colspan='2' class='sha1'><a href='/software/ldetect-lst/commit/lst/pcitable?h=V0_1_9_18mdk&amp;id=cfb87aeb6774a896879d7e139e2ab1823a23777b'>cfb87aeb6774a896879d7e139e2ab1823a23777b</a> (<a href='/software/ldetect-lst/diff/lst/pcitable?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d&amp;id2=cfb87aeb6774a896879d7e139e2ab1823a23777b'>diff</a>)</td></tr><tr><th>download</th><td colspan='2' class='sha1'><a href='/software/ldetect-lst/snapshot/ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar'>ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar</a><br/><a href='/software/ldetect-lst/snapshot/ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar.gz'>ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar.gz</a><br/><a href='/software/ldetect-lst/snapshot/ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar.bz2'>ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar.bz2</a><br/><a href='/software/ldetect-lst/snapshot/ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar.xz'>ldetect-lst-45fde6767af82053307942600680f19bd848a27d.tar.xz</a><br/><a href='/software/ldetect-lst/snapshot/ldetect-lst-45fde6767af82053307942600680f19bd848a27d.zip'>ldetect-lst-45fde6767af82053307942600680f19bd848a27d.zip</a><br/></td></tr></table>
<div class='commit-subject'>merge kernel-2.6.2.1mdk pcimap</div><div class='commit-msg'></div><div class='diffstat-header'><a href='/software/ldetect-lst/diff/?h=V0_1_9_18mdk&amp;id=45fde6767af82053307942600680f19bd848a27d'>Diffstat</a> (limited to 'lst/pcitable')</div><table summary='diffstat' class='diffstat'>