From b9b8b8286dde5918e8eae25dbeef79446b12e636 Mon Sep 17 00:00:00 2001
From: vallard <vallard@8638fb3e-16cb-4fca-ae20-7b5d299a9bcd>
Date: Thu, 12 Feb 2009 20:41:08 +0000
Subject: [PATCH] git-svn-id:
 https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@2735
 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd

---
 xCAT-server/lib/xcat/plugins/partimage.pm | 327 ++++++++++++++++++++++
 1 file changed, 327 insertions(+)
 create mode 100644 xCAT-server/lib/xcat/plugins/partimage.pm

diff --git a/xCAT-server/lib/xcat/plugins/partimage.pm b/xCAT-server/lib/xcat/plugins/partimage.pm
new file mode 100644
index 000000000..e2c1df7e5
--- /dev/null
+++ b/xCAT-server/lib/xcat/plugins/partimage.pm
@@ -0,0 +1,327 @@
+# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
+package xCAT_plugin::partimage;
+BEGIN
+{
+  $::XCATROOT = $ENV{'XCATROOT'} ? $ENV{'XCATROOT'} : '/opt/xcat';
+}
+use lib "$::XCATROOT/lib/perl";
+use Storable qw(dclone);
+use Sys::Syslog;
+use File::Temp qw/tempdir/;
+use xCAT::Table;
+use xCAT::Utils;
+use xCAT::MsgUtils;
+use xCAT::Template;
+use xCAT::Postage;
+use Data::Dumper;
+use Getopt::Long;
+Getopt::Long::Configure("bundling");
+Getopt::Long::Configure("pass_through");
+use File::Path;
+use File::Copy;
+
+my $pImageDir = "$::XCATROOT/share/xcat/install/partimage";
+my $tftpDir = "/tftpboot";
+my @cpiopid;
+
+sub handled_commands
+{
+	return {
+		mkinstall => "nodetype:os=imagecapture|imagerestore",
+	};
+}
+
+sub process_request
+{
+	my $request  = shift;
+	my $callback = shift;
+	my $doreq    = shift;
+	my $distname = undef;
+	my $arch     = undef;
+	my $path     = undef;
+	if($request->{command}->[0] eq 'mkinstall'){
+		return mkinstall($request, $callback, $doreq);
+  }else{
+		errorSig($callback, "Error in partimage plugin, shouldn't call anything other than mkinstall function");
+	}
+}
+
+
+sub errorSig {
+	$callback = shift;
+	$msg = shift;
+	$callback->({
+		error => ["$msg"],
+		errorcode => [1]
+	});
+}
+
+# function used for creating boot attributes for partimage functionality.
+sub mkinstall {
+	# we should be passed in the request which has the list of nodes
+	my $request  = shift;
+	# The callback function for what to print out and do with results.
+	my $callback = shift;
+	# doreq isn't used here, but we shift for good measure.
+	my $doreq    = shift;
+	my @nodes    = @{$request->{node}};
+	my $node;
+	# read the nodetype table for our nodes.
+	my $ostab = xCAT::Table->new('nodetype');
+	my %doneimgs;
+	my $mode;
+
+	# go through each node:
+	# 1. Verify that nodetype table is filled out and accurate
+	# 2. Create auto file by subbing through variables.
+	foreach $node (@nodes){
+		my $osinst;
+		my $ent = $ostab->getNodeAttribs($node, ['profile', 'os', 'arch']);
+		# Verify that the user configured the tables like we expect them to.
+		# if not, then give them a stern warning.
+		unless ($ent->{os}){
+			errorSig($callback, "No OS defined in nodetype for $node.  OS types for partimage should be imagecapture or imagerestore");
+			next;
+		}
+		unless($ent->{arch}){
+			errorSig($callback, "No arch defined in nodetype for $node. only x86 and x86_64 supported");
+			next;
+		}
+		unless($ent->{profile}){
+			errorSig($callback, "No profile defined in nodetype for $node.  Default profile types are found in $pImageDir");
+			next;
+		}
+		my $os      = $ent->{os};
+		my $arch    = $ent->{arch};
+		my $profile = $ent->{profile};
+
+		if($os eq "imagecapture"){
+			$mode = "capture";
+		}
+		elsif($os eq "imagerestore"){
+			$mode = "restore";
+		}else{
+			errorSig($callback, "Unrecognized mode: $os.  Should be imagecapture or imagerestore for partimage");
+			next;
+		}
+
+		# This is step 2. Take the template file and sub in the variables
+		# so that there is a file in /install/autoinst/$node for each node
+		my $tmplfile=get_tmpl_file_name("/install/custom/install/partimage", $profile, $os, $arch);
+		if (! $tmplfile) { $tmplfile=get_tmpl_file_name($pImageDir, $profile, $os, $arch); }
+		unless ( -r "$tmplfile"){
+			errorSig($callback, "No profile template exists for $profile");
+			next;
+		}
+
+		my $instDir = xCAT::Utils->get_site_attribute("installdir");
+		if(! defined($instDir)){
+			$instDir = "/install";
+		}
+
+		#Call the Template class to do substitution to produce the file
+		my $tmperr = xCAT::Template->subvars(
+			$tmplfile,
+			"$instDir/autoinst/$node",
+			$node
+		);
+		# if there were problems during substitution then let the user know
+		# about it.
+		if ($tmperr){
+			$callback->({
+				node => [{
+					name      => [$node],
+					error     => [$tmperr],
+					errorcode => [1]
+				}]
+			});
+			next;
+		}
+	
+		# part 3.  Create the image capture/restore image
+		# this is just a repackaging of the existing xCAT nbfs
+		# partimage uses the 32 bit one.  No problems if its 64 bit
+		# the 32 bit will still work.  This way we can capture 64 bit
+		# and 64 bit os's.  What a clever tool!
+
+		unless ($doneimgs{"$os|$arch"}){
+			mkpath("$tftpDir/xcat/partimage/x86");
+			mkpath("$instDir/partimage");
+
+			# verify ramdisk is there
+			if(-r "$tftpDir/xcat/nbfs.x86.gz"){
+				copy("$tftpDir/xcat/nbfs.x86.gz", "$tftpDir/xcat/partimage/x86/");
+			}else{
+				errorSig($callback, "$tftpDir/xcat/nbfs.x86.gz was not found, please run 'mknb x86' first.");
+			}
+
+			# verify that kernel is there
+			if(-r "$tftpDir/xcat/nbk.x86"){
+				copy("$tftpDir/xcat/nbk.x86", "$tftpDir/xcat/partimage/x86/");
+			}else{
+				errorSig($callback, "$tftpDir/xcat/nbk.x86 was not found, please run 'mknb x86' first.");
+			}
+
+			# now get the image supped up
+			mkPartImage($node,$tftpDir,$callback);
+			$doneimgs{"$os|$arch"} = 1;
+		}
+
+		# part 4.  Create the boot parameters for the machine.
+		#
+		# if there was an error then setBootParams will return 1;
+		# if there is no error it will return 0.
+		if(setBootParams($node,$callback, $instDir, $mode)){
+			next;
+		}
+	}		
+}
+
+
+sub mkPartImage{
+	# modify nbfs.tgz 
+	my $node = shift;
+	my $tftpDir = shift;
+	my $callback = shift;
+	my $pDataDir = "$pImageDir/data";
+	my $pNbDir = "$tftpDir/xcat/partimage/x86/tmp";
+# $callback->({data=>["Creating nbfs.$arch.gz in $tftpdir/xcat"]});
+	print "$pNbDir";
+	# remove this directory if it exists.
+	if(-d $pNbDir){
+		rmtree("$pNbDir");
+		#	errorSig($callback, "could not remove $pNbDir");
+	}
+	if(-r "$pNbDir/../pnbfs.x86.gz"){
+		unless(unlink("$pNbDir/../pnbfs.x86.gz") == 0 ){
+			errorSig($callback, "could not remove $pNbDir/../pnbfs.x86.gz\n");
+		}
+	}
+	mkpath($pNbDir);
+	$callback=>({data=>["cd $pNbDir; gunzip -c ../nbfs.x86.gz | cpio -id"]});	
+	system("cd $pNbDir; gunzip -c ../nbfs.x86.gz | cpio -id");	
+
+
+# Tput files
+# /usr/bin/tput
+	copy("$pDataDir/tput", "$pNbDir/usr/bin/");
+	system("chmod 755 $pNbDir/usr/bin/");
+	
+# /lib/libdl.so.2
+	copy("$pDataDir/lib/libdl.so.2", "$pNbDir/lib/");
+# /usr/lib/libncursesw.so.5
+	copy("$pDataDir/lib/libncursesw.so.5", "$pNbDir/usr/lib/");
+
+
+	copy("$pDataDir/lib/ld-linux.so.2", "$pNbDir/lib/");
+	system("chmod 755 $pNbDir/lib/ld-linux.so.2");
+
+	copy("$pDataDir/lib/libc.so.6", "$pNbDir/lib/");
+	system("chmod 755 $pNbDir/lib/libc.so.6");
+
+	copy("$pDataDir/sfdisk", "$pNbDir/sbin");
+	system("chmod 755 $pNbDir/sbin/sfdisk");
+
+	copy("$pDataDir/e2fsck", "$pNbDir/sbin");
+	system("chmod 755 $pNbDir/sbin/e2fsck");
+
+	copy("$pDataDir/partimage", "$pNbDir/bin/");
+	system("chmod 755 $pNbDir/bin/partimage");
+
+	copy("$pDataDir/partimage.sh", "$pNbDir/etc/init.d/S80partimage.sh");
+
+	unlink("$pNbDir/etc/init.d/S99xcat.sh");
+
+	# we make a new nbfs.x86.gz and call it pnbfs.  Its really just
+	# nbfs with some other things added and taken away.
+	system("cd $pNbDir; find . | cpio -o -H newc | gzip -9 > $pNbDir/../pnbfs.x86.gz");	
+
+	# remove all the junk after compressing the file.
+	rmtree("$pNbDir");
+
+}
+
+# set the boot parameters of the node.  If we return 1 there is an error
+# if we return 0 everything went according to plan.
+sub setBootParams{
+	my $node = shift;
+	my $callback = shift;
+	my $instDir = shift;
+	my $mode = shift;
+	# get all the info we need for the node:
+	# ms, NIC, etc.
+	my $restab = xCAT::Table->new('noderes');
+	my $bptab = xCAT::Table->new('bootparams',-create=>1);
+	my $hmtab  = xCAT::Table->new('nodehm');
+	my $ent    =
+		$restab->getNodeAttribs(
+			$node,['nfsserver', 'primarynic', 'installnic']
+		);
+	my $sent =
+		$hmtab->getNodeAttribs(
+			$node, ['serialport', 'serialspeed', 'serialflow']
+		);
+	unless ($ent and $ent->{nfsserver}){
+		errorSig($callback, "No noderes.nfsserver for $node defined");
+    $callback->("No noderes.nfsserver for $node defined");
+    return 1;
+	}
+
+	my $kernel = "xcat/partimage/x86/nbk.x86";
+	my $initrd = "xcat/partimage/x86/pnbfs.x86.gz";
+	my $kcmdline = "quiet";
+
+	# console stuff
+	if (defined $sent->{serialport}){
+		unless ($sent->{serialspeed}){
+			errorSig($callback, "serialport defined, but no serialspeed for $node in nodehm table");
+			return 1;
+		}
+		$kcmdline .=
+			" console=ttyS"
+			. $sent->{serialport} . ","
+			. $sent->{serialspeed};
+		if($sent and ($sent->{serialflow} =~ /(ctsrts|cts|hard)/)){
+			$kcmdline .= "n8r";
+		}
+	}
+
+	# xcatd stuff	
+	# get xCATd port:
+	my $xPort = xCAT::Utils->get_site_attribute("xcatdport");
+	$kcmdline .= " xcatd=" . $ent->{nfsserver} . ":" . $xPort;
+	$kcmdline .= " pcfg=http://" . $ent->{nfsserver} . "/$instDir/autoinst/$node";
+	$kcmdline .= " mode=$mode"; 
+	$bptab->setNodeAttribs(
+		$node, {
+			kernel   => $kernel,
+			initrd   => $initrd,
+			kcmdline => $kcmdline
+		}
+	);
+	return 0;
+}
+
+
+sub get_tmpl_file_name {
+  my $base=shift;
+  my $profile=shift;
+  my $os=shift;
+  my $arch=shift;
+  if (-r   "$base/$profile.$os.$arch.tmpl") {
+    return "$base/$profile.$os.$arch.tmpl";
+  }
+  elsif (-r "$base/$profile.$arch.tmpl") {
+    return  "$base/$profile.$arch.tmpl";
+  }
+  elsif (-r "$base/$profile.$os.tmpl") {
+    return  "$base/$profile.$os.tmpl";
+  }
+  elsif (-r "$base/$profile.tmpl") {
+    return  "$base/$profile.tmpl";
+  }
+
+  return "";
+}
+
+1;