9502050fec
git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@2158 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
255 lines
9.1 KiB
Perl
255 lines
9.1 KiB
Perl
package xCAT_plugin::packimage;
|
|
use xCAT::Table;
|
|
use Getopt::Long;
|
|
use File::Path;
|
|
use File::Copy;
|
|
use Cwd;
|
|
use File::Temp;
|
|
use xCAT::Utils qw(genpassword);
|
|
Getopt::Long::Configure("bundling");
|
|
Getopt::Long::Configure("pass_through");
|
|
|
|
sub handled_commands {
|
|
return {
|
|
packimage => "packimage",
|
|
}
|
|
}
|
|
|
|
sub process_request {
|
|
my $sitetab = xCAT::Table->new('site');
|
|
my $request = shift;
|
|
my $callback = shift;
|
|
my $doreq = shift;
|
|
my $ent = $sitetab->getAttribs({key=>'installdir'},['value']);
|
|
my $installroot = "/install";
|
|
|
|
if ($ent and $ent->{value}) {
|
|
$installroot = $ent->{value};
|
|
}
|
|
@ARGV = @{$request->{arg}};
|
|
my $osver;
|
|
my $arch;
|
|
my $profile;
|
|
my $method='cpio';
|
|
GetOptions(
|
|
"profile|p=s" => \$profile,
|
|
"arch|a=s" => \$arch,
|
|
"osver|o=s" => \$osver,
|
|
"method|m=s" => \$method,
|
|
"help|h" => \$help,
|
|
"version|v" => \$version
|
|
);
|
|
if ($version) {
|
|
my $version = xCAT::Utils->Version();
|
|
$callback->({info=>[$version]});
|
|
return;
|
|
}
|
|
if ($help) {
|
|
$callback->({info=>["packimage -h \npackimage -v \npackimage [-p profile] [-a architecture] [-o OS] [-m method]\n"]});
|
|
return;
|
|
}
|
|
my $distname = $osver;
|
|
until (-r "$::XCATROOT/share/xcat/netboot/$distname/" or not $distname) {
|
|
chop($distname);
|
|
}
|
|
unless ($distname) {
|
|
$callback->({error=>["Unable to find $::XCATROOT/share/xcat/netboot directory for $osver"],errorcode=>[1]});
|
|
return;
|
|
}
|
|
unless ($installroot) {
|
|
$callback->({error=>["No installdir defined in site table"],errorcode=>[1]});
|
|
return;
|
|
}
|
|
my $oldpath=cwd();
|
|
my $exlistloc=get_exlist_file_name("$installroot/custom/netboot/$distname", $profile, $osver, $arch);
|
|
if (!$exlistloc) { $exlistloc=get_exlist_file_name("$::XCATROOT/share/xcat/netboot/$distname", $profile, $osver, $arch); }
|
|
|
|
if (!$exlistloc)
|
|
{
|
|
$callback->({error=>["Unable to finde file exclusion list under $installroot/custom/netboot/$distname or $::XCATROOT/share/xcat/netboot/$distname/ for $profile/$arch/$osver"],errorcode=>[1]});
|
|
next;
|
|
}
|
|
#print "exlistloc=$exlistloc\n";
|
|
my $exlist;
|
|
open($exlist,"<",$exlistloc);
|
|
my $excludestr = "find . ";
|
|
while (<$exlist>) {
|
|
chomp $_;
|
|
s/\s*#.*//; #-- remove comments
|
|
next if /^\s*$/; #-- skip empty lines
|
|
$excludestr .= "'!' -path '".$_."' -a ";
|
|
}
|
|
close($exlist);
|
|
|
|
# add the xCAT post scripts to the image
|
|
if (! -d "$installroot/netboot/$osver/$arch/$profile/rootimg") {
|
|
$callback->({error=>["$installroot/netboot/$osver/$arch/$profile/rootimg does not exist, run genimage -o $osver -p $profile on a server with matching architecture"]});
|
|
return;
|
|
}
|
|
copybootscript($installroot, $osver, $arch, $profile, $callback);
|
|
my $passtab = xCAT::Table->new('passwd');
|
|
if ($passtab) {
|
|
(my $pent) = $passtab->getAttribs({key=>'system',username=>'root'},'password');
|
|
if ($pent and defined ($pent->{password})) {
|
|
my $pass = $pent->{password};
|
|
my $shadow;
|
|
open($shadow,"<","$installroot/netboot/$osver/$arch/$profile/rootimg/etc/shadow");
|
|
my @shadents = <$shadow>;
|
|
close($shadow);
|
|
open($shadow,">","$installroot/netboot/$osver/$arch/$profile/rootimg/etc/shadow");
|
|
unless ($pass =~ /^\$1\$/) {
|
|
$pass = crypt($pass,'$1$'.genpassword(8));
|
|
}
|
|
print $shadow "root:$pass:13880:0:99999:7:::\n";
|
|
foreach (@shadents) {
|
|
unless (/^root:/) {
|
|
print $shadow "$_";
|
|
}
|
|
}
|
|
close($shadow);
|
|
}
|
|
}
|
|
|
|
|
|
my $verb = "Packing";
|
|
if ($method =~ /nfs/) {
|
|
$verb = "Prepping";
|
|
}
|
|
if ($method =~ /nfs/) {
|
|
$callback->({data=>["\nNOTE: Contents of $installroot/netboot/$osver/$arch/$profile/rootimg\nMUST be available on all service and management nodes and NFS exported."]});
|
|
}
|
|
my $temppath;
|
|
my $oldumask;
|
|
if (! -d "$installroot/netboot/$osver/$arch/$profile/rootimg") {
|
|
$callback->({error=>["$installroot/netboot/$osver/$arch/$profile/rootimg does not exist, run genimage -o $osver -p $profile on a server with matching architecture"]});
|
|
return;
|
|
}
|
|
$callback->({data=>["$verb contents of $installroot/netboot/$osver/$arch/$profile/rootimg"]});
|
|
unlink("$installroot/netboot/$osver/$arch/$profile/rootimg.gz");
|
|
unlink("$installroot/netboot/$osver/$arch/$profile/rootimg.sfs");
|
|
unlink("$installroot/netboot/$osver/$arch/$profile/rootimg.nfs");
|
|
if ($method =~ /cpio/) {
|
|
$excludestr =~ s!-a \z!|cpio -H newc -o | gzip -c - > ../rootimg.gz!;
|
|
$oldmask = umask 0077;
|
|
} elsif ($method =~ /squashfs/) {
|
|
$temppath = mkdtemp("/tmp/packimage.$$.XXXXXXXX");
|
|
$excludestr =~ s!-a \z!|cpio -dump $temppath!;
|
|
} elsif ($method =~ /nfs/) {
|
|
$excludestr = "touch ../rootimg.nfs";
|
|
} else {
|
|
$callback->({error=>["Invalid method '$method' requested"],errorcode=>[1]});
|
|
}
|
|
chdir("$installroot/netboot/$osver/$arch/$profile/rootimg");
|
|
system($excludestr);
|
|
if ($method =~ /cpio/) {
|
|
chmod 0644,"$installroot/netboot/$osver/$arch/$profile/rootimg.gz";
|
|
umask $oldmask;
|
|
} elsif ($method =~ /squashfs/) {
|
|
my $flags;
|
|
if ($arch =~ /x86/) {
|
|
$flags="-le";
|
|
} elsif ($arch =~ /ppc/) {
|
|
$flags="-be";
|
|
}
|
|
if (! -x "/sbin/mksquashfs") {
|
|
$callback->({error=>["mksquashfs not found, squashfs-tools rpm should be installed on the management node"],errorcode=>[1]});
|
|
return;
|
|
}
|
|
my $rc = system("mksquashfs $temppath ../rootimg.sfs $flags");
|
|
if ($rc) {
|
|
$callback->({error=>["mksquashfs could not be run successfully"],errorcode=>[1]});
|
|
return;
|
|
}
|
|
$rc = system("rm -rf $temppath");
|
|
if ($rc) {
|
|
$callback->({error=>["Failed to clean up temp space"],errorcode=>[1]});
|
|
return;
|
|
}
|
|
chmod(0644,"../rootimg.sfs");
|
|
}
|
|
chdir($oldpath);
|
|
}
|
|
|
|
###########################################################
|
|
#
|
|
# copybootscript - copy the xCAT diskless init scripts to the image
|
|
#
|
|
#############################################################
|
|
sub copybootscript {
|
|
|
|
my $installroot = shift;
|
|
my $osver = shift;
|
|
my $arch = shift;
|
|
my $profile = shift;
|
|
my $callback = shift;
|
|
|
|
|
|
if ( -f "$installroot/postscripts/xcatdsklspost") {
|
|
|
|
# copy the xCAT diskless post script to the image
|
|
mkpath("$installroot/netboot/$osver/$arch/$profile/rootimg/opt/xcat");
|
|
|
|
copy ("$installroot/postscripts/xcatdsklspost", "$installroot/netboot/$osver/$arch/$profile/rootimg/opt/xcat/xcatdsklspost");
|
|
|
|
chmod(0755,"$installroot/netboot/$osver/$arch/$profile/rootimg/opt/xcat/xcatdsklspost");
|
|
|
|
} else {
|
|
|
|
my $rsp;
|
|
push @{$rsp->{data}}, "Could not find the script $installroot/postscripts/xcatdsklspost.\n";
|
|
xCAT::MsgUtils->message("E", $rsp, $callback);
|
|
return 1;
|
|
}
|
|
if ( -f "$installroot/postscripts/xcatdsklspost.aix") {
|
|
copy ("$installroot/postscripts/xcatdsklspost.aix", "$installroot/netboot/$osver/$arch/$profile/rootimg/opt/xcat/xcatdsklspost.aix");
|
|
chmod(0755,"$installroot/netboot/$osver/$arch/$profile/rootimg/opt/xcat/xcatdsklspost.aix");
|
|
}
|
|
|
|
if ( -f "$installroot/postscripts/xcatpostinit") {
|
|
|
|
# copy the linux diskless init script to the image
|
|
# - & set the permissions
|
|
copy ("$installroot/postscripts/xcatpostinit","$installroot/netboot/$osver/$arch/$profile/rootimg/etc/init.d/xcatpostinit");
|
|
|
|
chmod(0755,"$installroot/netboot/$osver/$arch/$profile/rootimg/etc/init.d/xcatpostinit");
|
|
|
|
# run chkconfig
|
|
#my $chkcmd = "chroot $installroot/netboot/$osver/$arch/$profile/rootimg chkconfig --add xcatpostinit";
|
|
symlink "/etc/init.d/xcatpostinit","$installroot/netboot/$osver/$arch/$profile/rootimg/etc/rc3.d/S84xcatpostinit";
|
|
symlink "/etc/init.d/xcatpostinit","$installroot/netboot/$osver/$arch/$profile/rootimg/etc/rc4.d/S84xcatpostinit";
|
|
symlink "/etc/init.d/xcatpostinit","$installroot/netboot/$osver/$arch/$profile/rootimg/etc/rc5.d/S84xcatpostinit";
|
|
#my $rc = system($chkcmd);
|
|
#if ($rc) {
|
|
#my $rsp;
|
|
# push @{$rsp->{data}}, "Could not run the chkconfig command.\n";
|
|
# xCAT::MsgUtils->message("E", $rsp, $callback);
|
|
# return 1;
|
|
# }
|
|
} else {
|
|
my $rsp;
|
|
push @{$rsp->{data}}, "Could not find the script $installroot/postscripts/xcatpostinit.\n";
|
|
xCAT::MsgUtils->message("E", $rsp, $callback);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
sub get_exlist_file_name {
|
|
my $base=shift;
|
|
my $profile=shift;
|
|
my $osver=shift;
|
|
my $arch=shift;
|
|
|
|
my $exlistloc="";
|
|
if (-r "$base/$profile.$osver.$arch.exlist") {
|
|
$exlistloc = "$base/$profile.$osver.$arch.exlist";
|
|
} elsif (-r "$base/$profile.$arch.exlist") {
|
|
$exlistloc = "$base/$profile.$arch.exlist";
|
|
} elsif (-r "$base/$profile.$osver.exlist") {
|
|
$exlistloc = "$base/$profile.$osver.exlist";
|
|
} elsif (-r "$base/$profile.exlist") {
|
|
$exlistloc = "$base/$profile.exlist";
|
|
}
|
|
return $exlistloc;
|
|
}
|