Changeset 49 for devel/dploy-pxe/bin/dploy-add2pxe

Timestamp:

04/16/08 15:58:35 (4 years ago)

Author:

renaudga

Message:

Work done to pass parameters to aad2pxe. functions need to be writen.

File:

• devel/dploy-pxe/bin/dploy-add2pxe (modified) (5 diffs)

devel/dploy-pxe/bin/dploy-add2pxe

@@ -12 +12 @@
 use File::Basename;
 
+my %opts;
+my $type = "";
+my $mac = "";
+my $url = "";
+my $ksdevice = "eth0";
+my $prefix = "";
+my $mrserver = "";
+my $mrpath = "";
+
 GetOptions("help|?|h" => \$opts{'h'},
-           "mode|m=s" => \$opts{'mode'},    # Mode is MR or SD
-           "mac|m=s" => \$opts{'mac'},      # Mac address
-             "ip|i=s" => \$opts{'ip'},      # 
-        "netmask|n=s" => \$opts{'mask'},
-        "gw|g=s" => \$opts{'gw'},
+           "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
+        ) || dploy_syntax(-1,0);
 
+if (defined $opts{'type'}) {
+  $type = $opts{'type'};
+}
 
+if (defined $opts{'h'}) {
+  dploy_syntax(0,1);
+}
 
 # Root of the PXE setup
-my $pxe="/tftpboot/pxelinux.cfg";
-# Where to find the LinuxCOE created ISO on the SystemDesigner server
-my $linuxcoe="/var/ftp/html/scratch_monkey/systemdesigner";
-# Where to find the Mondorescue created ISO on the Images server
-my $mondorescue="/var/ftp/html/mondorescue/iso";
+my $pxe=dploy_get_conf("pathpxe");
+
 # temporary mountpoint
-my $mnt="/tmp/mnt-dploy";
+if (not defined $ENV {'TMPDIR'}) {
+    $ENV{'TMPDIR'} = "/tmp";
+}
+my $tmp=tempdir("dploy.XXXXXXXXXX", DIR => $ENV{'TMPDIR'}, CLEANUP => 1);
+my $mnt="$tmp/iso";
+my $mnt2="$tmp/nfs";
 
 my $mac=$ARGV[0] || "";
 
+# Check Mac consitency (to be done in an external function)
 if ($mac !~ /^(?:[[:xdigit:]]{1,2}[-:]){5}[[:xdigit:]]{1,2}$/) {
         print "Syntax: dploy-add2pxe MAC url-to-iso\n";
@@ -45 +62 @@
 $newmac =~ s/:/-/g;
 
-my $url=$ARGV[1];
+# Check URL syntax
+if (defined $opts{'url'}) {
+  $url = $opts{'url'};
+}
 
 if ($url !~ /(\w+):\/\/([^\/:]+)(:\d+)?\/(.*)/) {
@@ -51 +71 @@
     exit(-1);
 }
+
+# Get Mondorescue values
+$mrserver = dploy_get_url();
+$mrpath = dploy_get_url();
 
 # Get the ID given by LinuxCOE/MondoRescue to prepare kernel/initrd/ks
@@ -60 +84 @@
 my $ks=$url;
 $ks =~ s/img([0-9]+)\.iso/ks$1.cfg/;
+
+# mount the ISO to get kernel and initrd
+mkdir "$mnt",0755;
+if ($Type == "LinuxCOE") {
+    system("sudo /bin/mount -o loop $linuxcoe/$iso $mnt");
+} else {
+    system("sudo /bin/mount -t nfs $mrserver:$mrpath $mnt2");
+    system("sudo /bin/mount -o loop $mnt2/$prefix-1.iso $mnt");
+}
+# Install script will have to check (and create if not existing) /tftpboot/tmp
+copy("$mnt/isolinux/vmlinuz","$pxe/../tmp/$k") or die "Copy failed: $!";
+copy("$mnt/isolinux/initrd.img","$pxe/../tmp/$i") or die "Copy failed: $!";
+
+# Get the Command Line parameters for MondoRescue restoration
+open(ISOLINUX, "$mnt/isolinux.cfg") || die "Unable to open $mnt/isolinux.cfg for writing";
+while ( <ISOLINUX> ) {
+    if (/label nuke/) {
+        my $cliparams = <ISOLINUX>;     # kernel line
+        $cliparams = <ISOLINUX>;        # append line
+    }
+}
+$cliparams =~ s/initrd=[a-z.] /initrd=initrd=mondorescue\/$i /;
+
+close ISOLINUX;
+# All done with ISO image... umounting
+system("sudo /bin/umount $mnt");
+
 
 # Updates PXElinux
@@ -73 +124 @@
   localboot 0
 
-label linuxcoe
-  kernel linuxcoe/$k
-  append ksdevice=eth0 ks=$ks initrd=linuxcoe/$i text
 EOF
 close(PXE);
 
-# mount the ISO to get kernel and initrd
-mkdir "$mnt",0755;
-system("sudo /bin/mount -o loop $linuxcoe/$iso $mnt");
-copy("$mnt/isolinux/vmlinuz","$pxe/../linuxcoe/$k") or die "Copy failed: $!";
-copy("$mnt/isolinux/initrd.img","$pxe/../linuxcoe/$i") or die "Copy failed: $!";
-system("sudo /bin/umount $mnt");
+# Customize pxe file depending on Job type submited
+open(PXE, ">> $pxe/01-$newmac") || die "Unable to open $pxe/01-$newmac for writing";
+if ($Type == "LinuxCOE") {
+    print PXE "label linuxcoe\n";
+    print PXE "  kernel linuxcoe/$k\n";
+    print PXE "  append ksdevice=$ksdevice ks=$ks initrd=linuxcoe/$i text\n";
+} else {
+    print PXE "label mondorecue\n";
+    print PXE "  kernel mondorescue/$k\n";
+    print PXE "  append $cliparams pxe prefix=$prefix ipconf=$ksdevice:dhcp nfsmount=$mrserver:$mrpath\n";
+}
+close(PXE);
+