source: devel/dploy-pxe/bin/dploy-add2pxe @ 60

#!/usr/bin/perl -w
# 
# $Id$
# 
# (c) Bruno Cornec, HP
# (c) Gallig Renaud, HP
#
# Provided under the GPL v2

use strict;
use Pod::Usage;
use Getopt::Long qw(:config auto_abbrev no_ignore_case);
use Data::Dumper;
use File::Copy;
use File::Basename;
use ProjectBuilder::Base;
use ProjectBuilder::Conf;
use Dploy::Base;

=pod

=head1 NAME

dploy-add2pxe - Update PXE configuration file for dploy.org

=head1 DESCRIPTION

dploy-add2pxe helps you keeping your PXE configuration up to date 

=head1 SYNOPSIS

dploy-add2pxe -t mondorescue|linuxcoe -m MAC-address -u ISO-image-URL

dploy-add2pxe --type mondorescue|linuxcoe --mac MAC-address --url ISO-image-URL

=head1 OPTIONS

=over 4

=item B<-h|--help>

Print a brief help message and exits.

=cut


my %opts;                                       # GetOpt Hash
my $url = undef;                        # url of the boot ISO image
my $ksdevice = "eth0";          # NIC device to boot on
my $prefix = undef;                     # Prefix of the MondoRescue image
my $cliparams;                          # Command Line parames of the MondoRescue image
my $destdir = undef;            # Destination driectory for tftp files

pb_syntax_init("dploy-add2pxe Version PBVER-PBREV\n");

GetOptions("help|?|h" => \$opts{'h'},
           "type|t=s" => \$opts{'type'},        # Type is MondoRescue or LinuxCOE
            "mac|m=s" => \$opts{'mac'},         # Mac address
            "url|u=s" => \$opts{'url'},         # URL of ISO image containing kernel, modules and answer file
                ) || pb_syntax(-1,0);

if (defined $opts{'h'}) {
    pb_syntax(0,1);
}

# type is always lowercase
$opts{'type'} = lc($opts{'type'});

my $dploydir = "PBCONF";
pb_conf_add("$dploydir/dploy.conf");

# Root of the PXE setup
my ($pxe)=pb_conf_get("pathpxe");
my $pathpxe = $pxe->{"default"};

# temporary mountpoint
pb_temp_init();
my $mnt="$ENV{'PBTMP'}/iso";
my $mnt2="$ENV{'PBTMP'}/nfs";

# Check Mac consitency 
my $mac = dploy_check_mac($opts{'mac'});

my ($scheme, $account, $mrserver, $port, $mrpath) = pb_get_uri($opts{'url'});

# Get the ID given by LinuxCOE/MondoRescue to prepare kernel/initrd/ks
my $iso=basename($opts{'url'});
my $id=$iso;
$id =~ s/img([0-9]+)\.iso/$1/;
my $k="k-$id";
my $i="i-$id";
my $ks="ks-$id.cfg";

# mount the ISO to get kernel and initrd
mkdir "$mnt",0755;
if ($opts{'type'} =~ /linuxcoe/) {
        my $bn = basename($mrpath);
        unlink("$ENV{'PBTMP'}/$bn");
        pb_system("cd \$PBTMP ; wget $opts{'url'}","Downloading $opts{'url'}");
        pb_system("sudo /bin/mount -o loop $ENV{'PBTMP'}/$bn $mnt","Mounting $bn");
} elsif ($opts{'type'} =~ /mondorescue/) {
        pb_system("sudo /bin/mount -t nfs $mrserver:$mrpath $mnt2","Mountig NFS $mrserver:$mrpath");
        pb_system("sudo /bin/mount -o loop $mnt2/$prefix-1.iso $mnt","Mounting LOOP $mnt2/$prefix-1.iso");
} else {
    # Type passed has parameter does not contain a valid value
    pb_syntax(-1.0);
}

# Install script will have to check (and create if not existing) /tftpboot/tmp
# $pathpxe is normaly setup correctly by package install
$destdir = "$pathpxe/../$opts{'type'}";
copy("$mnt/isolinux/vmlinuz","$destdir/$k") or die "Copy of $mnt/isolinux/vmlinuz to $destdir/$k failed: $!";
copy("$mnt/isolinux/initrd.img","$destdir/$i") or die "Copy of $mnt/isolinux/initrd.img to $destdir/$i failed: $!";
if ($opts{'type'} =~ /linuxcoe/) {
        copy("$mnt/KS.CFG","$destdir/$ks") or die "Copy of $mnt/KS.CFG to $destdir/$ks failed: $!";
}

# Get the Command Line parameters for MondoRescue restoration
if ($opts{'type'} =~ /mondorescue/) {
        open(ISOLINUX, "$mnt/isolinux.cfg") || die "Unable to open $mnt/isolinux.cfg for writing";
        while ( <ISOLINUX> ) {
                if (/label nuke/) {
                        $cliparams = <ISOLINUX>;                # kernel line
                        $cliparams = <ISOLINUX>;                # append line
                }
        }
        $cliparams =~ s/initrd=[A-z.] /initrd=mondorescue\/$i /;
        close ISOLINUX;
}

# All done with ISO image... umounting
pb_system("sudo /bin/umount $mnt","Umount ISO image");
if ($opts{'type'} =~ /mondorescue/) {
        pb_system("sudo /bin/umount $mnt2","Unmountig NFS $mrserver:$mrpath");
}

# Updates PXElinux
open(PXE, "> $pathpxe/01-$mac") || die "Unable to open $pathpxe/01-$mac for writing";
print PXE << "EOF";
default local
prompt 1
timeout 600
display msg/linuxcoe.msg
allowoptions 1

label local
  localboot 0

EOF

# Customize pxe file depending on Job type submited
if ($opts{'type'} =~ /linuxcoe/) {
        print PXE << "EOF";
label linuxcoe
  kernel linuxcoe/$k
  append ksdevice=$ksdevice ks=$ks initrd=linuxcoe/$i text
EOF
} else {
        print PXE << "EOF";
label mondorescue
  kernel mondorescue/$k
  append $cliparams pxe prefix=$prefix ipconf=$ksdevice:dhcp nfsmount=$mrserver:$mrpath
EOF
}
close(PXE);