xcat-core/xCAT-server/lib/xcat/plugins/osdistro.pm
2012-08-30 10:06:40 +00:00

478 lines
8.9 KiB
Perl

# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
#-------------------------------------------------------
=head1
xCAT plugin package to handle osdistro management
Supported command:
rmosdistro->rmosdistro
=cut
#-------------------------------------------------------
package xCAT_plugin::osdistro;
use Sys::Hostname;
use xCAT::Table;
use xCAT::Utils;
use xCAT::MsgUtils;
use Getopt::Long;
my $SUB_REQ_L;
1;
#-------------------------------------------------------
=head3 handled_commands
Return list of commands handled by this plugin
=cut
#-------------------------------------------------------
sub handled_commands
{
return {
mkosdistro => "osdistro",
lsosdistro => "osdistro",
rmosdistro => "osdistro",
};
}
#-------------------------------------------------------
=head3 process_request
Process the command
=cut
#-------------------------------------------------------
sub process_request
{
my $request = shift;
my $callback = shift;
$SUB_REQ_L = shift;
my $command = $request->{command}->[0];
my $args = $request->{arg};
my $envs = $request->{env};
my %rsp;
# do your processing here
# return info
if($command eq "mkosdistro")
{
&mkosdistro($request,$callback);
}elsif($command eq "lsosdistro")
{
&lsosdistro($request,$callback);
}elsif($command eq "rmosdistro")
{
&rmosdistro($request,$callback);
}
return;
}
sub parseosver
{
my $osver=shift;
if($osver =~ (/(\D+)(\d*)\.*(\d*)/))
{
return ($1,$2,$3);
}
return ();
}
#-------------------------------------------------------
=head3 mkosdistro
add/set osdistro entry
=cut
#-------------------------------------------------------
sub mkosdistro
{
my $request = shift;
my $callback = shift;
my %keyhash=();
my %updates=();
my $distname=undef;
my $osver=undef;
my $arch=undef;
my $help=undef;
my $force=undef;
my $path=undef;
if ($request->{arg}) {
@ARGV = @{$request->{arg}};
}
GetOptions(
'n|name=s' => \$distname,
'o|osver=s' =>\$osver,
'a|arch=s' => \$arch,
'h|help' => \$help,
'f|force' => \$force,
'p|path=s' => \$path
);
if ($help) {
$callback->({info=>"mkosdistro [{-n|--name}=osdistroname] [{-o|--osver}=osver] [{-a|--arch}=architecture] [{-p|--path}=ospkgpath] [-f|--force]",errorcode=>[0]});
return;
}
unless ($distname) {
$callback->({error=>"mkosdistro: osdistroname must be specified!",errorcode=>[1]});
return;
}
$keyhash{osdistroname} = $distname;
if($arch){
$updates{arch} = $arch;
}
if($osver){
($updates{basename},$updates{majorversion},$updates{minorversion}) = &parseosver($osver);
$updates{type} ="linux";
}
my $tab = xCAT::Table->new('osdistro',-create=>1);
unless($tab)
{
$callback->({error=>"mkosdistro: failed to open table 'osdistro'!",errorcode=>[1]});
return;
}
if($path)
{
$path =~ s/\/+$//;
(my $ref) = $tab->getAttribs(\%keyhash, 'dirpaths');
if ($ref and $ref->{dirpaths} )
{
unless($ref->{dirpaths} =~ /^($path)\/*,|,($path)\/*,|,($path)\/*$|^($path)\/*$/)
{
$updates{dirpaths}=$ref->{dirpaths}.','.$path;
}
}else
{
$updates{dirpaths} = $path;
}
}
if(%updates)
{
$tab->setAttribs( \%keyhash,\%updates );
$tab->commit;
}
$tab->close;
$callback->({info=>"mkosdistro: $distname success",errorcode=>[0]});
return;
}
#-------------------------------------------------------
=head3 getOSdistroref
get the refernece of osdistro by osimage
=cut
#-------------------------------------------------------
sub getOSdistroref
{
my $osimagetab=shift;
my $osdistroname=shift;
my $ret=();
unless($osimagetab)
{
return undef;
}
my @clause=();
push(@clause,"osdistroname==".$osdistroname);
my @result=$osimagetab->getAllAttribsWhere(\@clause,'imagename');
if(scalar @result)
{
foreach(@result)
{
$ret=$ret.$_->{'imagename'}.",";
}
}
else
{
return undef;
}
return $ret;
}
#-------------------------------------------------------
=head3 rmosdistro
remove osdistro entry
=cut
#-------------------------------------------------------
sub rmosdistro
{
my $request = shift;
my $callback = shift;
my %keyhash=();
my $distname=undef;
my $force=undef;
my $help=undef;
my $osdistropath=undef;
my @OSdistroListToDel=();
if ($request->{arg}) {
@ARGV = @{$request->{arg}};
}
GetOptions(
'h|help' => \$help,
'n|name=s' => \$distname,
'f|force' => \$force
);
if ($help) {
$callback->({info=>"rmosdistro [{-n|--name}=osdistroname] [-f|--force]",errorcode=>[0]});
return;
}
my $osdistrotab = xCAT::Table->new('osdistro',-create=>1);
unless($osdistrotab)
{
$callback->({error=>"rmosdistro: failed to open table 'osdistro'!",errorcode=>[1]});
return;
}
if($distname)
{
push(@OSdistroListToDel,$distname);
}
else
{
my @result=$osdistrotab->getAllAttribs('osdistroname');
if(defined(@result) and scalar @result >0)
{
foreach(@result)
{
push(@OSdistroListToDel,$_->{'osdistroname'});
}
}
}
if(scalar @OSdistroListToDel)
{
my $osimagetab=undef;
unless($force)
{
$osimagetab=xCAT::Table->new('osimage');
unless($osimagetab)
{
$callback->({error=>"rmosdistro: failed to open table 'osimage'!",errorcode=>[1]});
return;
}
}
foreach(@OSdistroListToDel)
{
unless($force)
{
my $result=&getOSdistroref($osimagetab,$_);
if($result)
{
$callback->({error=>"rmosdistro: failed to remove $_, it is referenced by osimages:$result, retry with -f option !",errorcode=>[1]});
next;
}
}
$keyhash{osdistroname} = $_;
my $result=$osdistrotab->getAttribs(\%keyhash,'dirpaths');
unless($result)
{
$callback->({error=>"rmosdistro: $keyhash{osdistroname} not exist!",errorcode=>[1]});
next;
}
if($result->{'dirpaths'})
{
$result->{'dirpaths'} =~ s/,/\ /g;
#$callback->({error=>"rmosdistro: remove $result->{'dirpaths'} directory!",errorcode=>[0]});
system("rm -rf $result->{'dirpaths'}");
if($? != 0)
{
$callback->({error=>"rmosdistro: failed to remove $keyhash{osdistroname} directory!",errorcode=>[1]});
next;
}
}
$osdistrotab->delEntries(\%keyhash);
$osdistrotab->commit;
$callback->({info=>"rmosdistro: remove $_ success",errorcode=>[0]});
}
if($osimagetab)
{
$osimagetab->close;
}
}
$osdistrotab->close;
return;
}
#-------------------------------------------------------
=head3 lsosdistro
list osdistro entry
=cut
#-------------------------------------------------------
sub lsosdistro
{
my $request = shift;
my $callback = shift;
my @clause=();
my @result=();
my $result;
my $distname=undef;
my $basename=undef;
my $majorversion=undef;
my $minorversion=undef;
my $osver=undef;
my $arch=undef;
my $type=undef;
my $help=undef;
my $stanza=undef;
if ($request->{arg}) {
@ARGV = @{$request->{arg}};
}
GetOptions(
'n|name=s' => \$distname,
'o|osver=s' =>\$osver,
'a|arch=s' => \$arch,
't|type=s' => \$type,
'h|help' => \$help,
'z|stanza' => \$stanza,
);
if ($help) {
$callback->({info=>"lsosdistro [{-n|--name}=osdistroname] [{-o|--osver}=osver] [{-t|--type}=ostype] [{-a|--arch}=architecture][-z|--stanza]",errorcode=>[0]});
return;
}
if($distname)
{
push(@clause, '-w');
push(@clause,"osdistroname==".$distname);
}
if($arch)
{
push(@clause, '-w');
push(@clause,"arch==".$arch);
}
if($type)
{
push(@clause, '-w');
push(@clause,"type==".$type);
}
if($osver)
{
($basename,$majorversion,$minorversion) = &parseosver($osver);
if($basename)
{
push(@clause, '-w');
push(@clause,"basename==".$basename);
}
if($majorversion)
{
push(@clause, '-w');
push(@clause,"majorversion==".$majorversion);
}
if($minorversion)
{
push(@clause, '-w');
push(@clause,"minorversion==".$minorversion);
}
}
push(@clause,"osdistro");
my $ret = xCAT::Utils->runxcmd({ command => ['tabdump'], arg => \@clause}, $SUB_REQ_L, 0, 1);
#print "xxxxx $$ret[1]";
foreach my $line(@$ret[1..$#$ret])
{
$result=$result.$line."\n";
}
$callback->({info=>"$result",errorcode=>[0]});
return;
}