console_reset.in 2.01 KB
Newer Older
1
#!/usr/bin/perl -wT
Leigh Stoller's avatar
Leigh Stoller committed
2 3 4

#
# Copyright (c) 2000-2002 University of Utah and the Flux Group.
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
# 
# {{{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/>.
# 
# }}}
Leigh Stoller's avatar
Leigh Stoller committed
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
use English;

#
# usage: console_reset
#
sub usage()
{
    print STDOUT "Usage: console_reset\n";
    exit(-1);
}

#
# Configure variables
#
my $TB		= "@prefix@";

#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use libdb;
use libtestbed;

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

my $consetup = "$TB/libexec/console_setup";
my @row;
my @nodes    = ();
my %shelves  = ();

# un-taint path
$ENV{'PATH'} = '/bin:/usr/bin:/usr/local/bin';
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

if (@ARGV) {
    usage();
}

#
# Only root or admins should run this script.
#
if ($UID && !TBAdmin($UID)) {
    die("Only root or TB admins can run this script.");

}

#
# Get a list of all of the experimental nodes.
#
my $query_result =
Leigh Stoller's avatar
Leigh Stoller committed
77 78
    DBQueryFatal("select node_id from nodes where role='testnode' ".
		 "order by priority");
79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104

while (@row = $query_result->fetchrow_array()) {
    my $node = $row[0];
    push(@nodes, $node);
}

#
# Issue small batches of setup.
#
while (@nodes) {
    my @batch = ();
    my $i = 0;

    while ($i < 10 && @nodes > 0) {
	my $node = shift(@nodes);
	
	push(@batch, $node);
	$i++;
    }
    
    print "Issuing console_setup for @batch ...\n";

    system("$consetup @batch");
}

exit 0;