xcat-core/xCAT-server-2.0/lib/xcat/plugins/TFTPsn.pm
2008-03-16 13:40:12 +00:00

160 lines
3.9 KiB
Perl

#!/usr/bin/env perl
# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
#-------------------------------------------------------
package xCAT_plugin::TFTPsn;
use xCAT::Table;
use xCAT::Utils;
use xCAT::MsgUtils;
use Getopt::Long;
#-------------------------------------------------------
=head1
xCAT plugin package to setup of atftp on both service node and MS
#-------------------------------------------------------
=head3 handled_commands
This runs on Service Node and on Management Server
Call setup_TFTP (actually setting up atftp)
=cut
#-------------------------------------------------------
sub handled_commands
{
my $rc = 0;
# setup atftp
my $service = "tftpserver";
$rc = &setup_TFTP(); # setup TFTP (ATFTP)
if ($rc == 0)
{
if (xCAT::Utils->isServiceNode())
{
xCAT::Utils->update_xCATSN($service);
}
}
return $rc;
}
#-------------------------------------------------------
=head3 process_request
Process the command
=cut
#-------------------------------------------------------
sub process_request
{
return;
}
#-----------------------------------------------------------------------------
=head3 setup_TFTP
Sets up TFTP services (using atftp)
=cut
#-----------------------------------------------------------------------------
sub setup_TFTP
{
my ($nodename) = @_;
my $rc = 0;
my $tftpdir = "/tftpboot"; # default
my $cmd;
my $master;
my $os;
my $arch;
$XCATROOT = "/opt/xcat"; # default
if ($ENV{'XCATROOT'})
{
$XCATROOT = $ENV{'XCATROOT'};
}
if (xCAT::Utils->isServiceNode())
{
# read DB for nodeinfo
my @nodeinfo = xCAT::Utils->determinehostname;
my $nodename = pop @nodeinfo; # get hostname
my @nodeipaddr = @nodeinfo; # get ip addresses
my $retdata = xCAT::Utils->readSNInfo($nodename);
$master = $retdata->{'master'};
$os = $retdata->{'os'};
$arch = $retdata->{'arch'};
if (!($arch))
{ # error
xCAT::MsgUtils->message("S", " Error reading service node arch.");
return 1;
}
}
# check to see if atftp is installed
$cmd = "/usr/sbin/in.tftpd -V";
my @output = xCAT::Utils->runcmd($cmd, -1);
if ($::RUNCMD_RC != 0)
{ # not installed
xCAT::MsgUtils->message("S", "atftp is not installed");
return 1;
}
if ($output[0] =~ "atftp") # it is atftp
{
# read tftp directory from database, if it exists
my @tftpdir1 = xCAT::Utils->get_site_attribute("tftpdir");
if ($tftpdir1[0])
{
$tftpdir = $tftpdir1[0];
}
mkdir($tftpdir);
# update fstab so that it will restart on reboot
if (xCAT::Utils->isServiceNode())
{
$cmd =
"fgrep \"$master:$tftpdir $tftpdir nfs timeo=14,intr 1 2\" /etc/fstab";
xCAT::Utils->runcmd($cmd, -1);
if ($::RUNCMD_RC != 0) # not already there
{
`echo "$master:$tftpdir $tftpdir nfs timeo=14,intr 1 2" >>/etc/fstab`;
}
}
# start atftp for both MS and Service node
$cmd = "service tftpd stop";
xCAT::Utils->runcmd($cmd, -1);
if ($::RUNCMD_RC != 0)
{
xCAT::MsgUtils->message("S", "Error from command:$cmd");
}
$cmd = "service tftpd start";
xCAT::Utils->runcmd($cmd, -1);
if ($::RUNCMD_RC != 0)
{
xCAT::MsgUtils->message("S", "Error from command:$cmd");
return 1;
}
}
else
{ # no ATFTP
xCAT::MsgUtils->message("S", "atftp is not installed");
return 1;
}
return $rc;
}
1;