power_ue.pm.in 2.11 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
#!/usr/bin/perl -wT

#
# Copyright (c) 2004 University of Utah and the Flux Group.
# 
# {{{EMULAB-LICENSE
# 
# This file is part of the Emulab network testbed software.
# 
# This file is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or (at
# your option) any later version.
# 
# This file is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public
# License for more details.
# 
# You should have received a copy of the GNU Affero General Public License
# along with this file.  If not, see <http://www.gnu.org/licenses/>.
# 
# }}}
#

# A little perl module to power cycle a ue attached to a console host -
# basically just a wrapper around the tbadb script.

package power_ue;

use Exporter;
@ISA = ("Exporter");
@EXPORT = qw( uectrl );

#
# Commands we run
#
my $TBROOT = '@prefix@';
my $TBADB = "$TBROOT/bin/tbadb";

# Turn off line buffering on output
$| = 1;

# usage: uectrl(cmd, devices)
# cmd = { "cycle" | "on" | "off" }
# devices = list of one or more physcial ue names.
#
# Returns 0 on success. Non-zero on failure.
# 
sub uectrl($@) {
    my ($cmd, @devices) = @_;
    my $tbadb_cmd;
    my $err = 0;

    #
    # Call TBADB as appropriate for device.  Note that we don't power down UE
    # nodes.  Instead, we reload them into "fastboot".
    #
    SWITCH: for ($cmd) {
	# Just reboot the device normally.  Presumably it is in
	# fastboot mode if we are "powering" it on.
	/^on$/i || /^cycle$/i and do {
	    $tbadb_cmd = "reboot";
	    last SWITCH;
	};
	# Put the device in "fastboot" mode.
	/^off$/i and do {
	    $tbadb_cmd = "reboot fastboot";
	    last SWITCH;
	};
	# DEFAULT
	print STDERR "power_ue: Unknown command: $cmd\n";
	return 1;
    }

    foreach my $dev (@devices) {
	if (system("$TBADB -n $dev $tbadb_cmd") != 0) {
	    print STDERR "power_ue: \"$cmd\" command failed for $dev\n";
	    $err = 1;
	}
    }

    return $err;
}

1;