node_exclude.in 2.38 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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
#!/usr/bin/perl -w
#
# Copyright (c) 2000-2014 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/>.
# 
# }}}
#
use English;
use Getopt::Std;
use strict;

#
# Exclude a node form protogeni.
#
sub usage()
{
    print "Usage: node_exclude [-r] ...\n";
    print "Options:\n";
    print " -r      Remove restriction\n";
    exit(-1);
}
my $optlist   = "r";
my $debug     = 0;
my $undo      = 0;

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

#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use libtestbed;
use emdb;
use EmulabConstants;
use Node;
my $TBOPSPID = EmulabConstants::TBOPSPID();

#
# Function prototypes
#
sub fatal($);

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

#
# Untaint the path
# 
$ENV{'PATH'} = "/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin";
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

#
# Parse command arguments. Once we return from getopts, all that should be
# left are the required arguments.
#
my %options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (defined($options{"r"})) {
    $undo = 1;
}
usage()
    if (@ARGV != 1);

my $node = Node->Lookup($ARGV[0]);
if (!defined($node)) {
    fatal("No such node!");
}
my $node_id = $node->node_id();

if ($undo) {
    DBQueryFatal("delete from node_attributes ".
		 "where node_id='$node_id' and ".
		 "      (attrkey='protogeni_exclude' or ".
		 "       attrkey='allowed_projects')");
}
else {
    DBQueryFatal("replace into node_attributes set ".
		 "    node_id='$node_id', ".
		 "    attrkey='protogeni_exclude', attrvalue='Yep'");
    DBQueryFatal("replace into node_attributes set ".
		 "    node_id='$node_id', ".
		 "    attrkey='allowed_projects', attrvalue='$TBOPSPID'");
}
exit(0);