setdest.in 2.6 KB
Newer Older
1 2 3
#!/usr/bin/perl -wT
#
# EMULAB-COPYRIGHT
Leigh B. Stoller's avatar
Leigh B. Stoller committed
4
# Copyright (c) 2005, 2007 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 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
# All rights reserved.
#
use English;
use Getopt::Std;

#
# Change delay params for a link.
#
sub usage()
{
    print(STDERR
	  "Usage: setdest -x xpos -y ypos -o orientation node_id\n");
    # Web interface cares about this return value!
    exit(-1);
}
my $optlist = "dx:y:o:";
my $debug   = 0;
my $destx;
my $desty;
my $desto;

#
# Configure variables
#
my $TB		= "@prefix@";
my $TBOPS       = "@TBOPSEMAIL@";
my $TEVC        = "$TB/bin/tevc";
my $nodeid;

#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use libdb;
use libtestbed;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
40
use Node;
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64

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

#
# Untaint the path
# 
$ENV{'PATH'} = "/bin:/sbin:/usr/bin:/usr/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.
#
%options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (defined($options{"d"})) {
    $debug = 1;
}
if (defined($options{"x"})) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
65
    $destx = $options{"x"};
66 67
}
if (defined($options{"y"})) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
68
    $desty = $options{"y"};
69 70
}
if (defined($options{"o"})) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
71
    $desto = $options{"o"};
72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
}
if (@ARGV != 1) {
    usage();
}
if (! (defined($destx) && defined($desty) && defined($desto))) {
    usage();
}
$nodeid = $ARGV[0];

#
# Untaint args.
#
if ($destx =~ /^([\d.]+)$/) {
    $destx = $1;
}
else {
88
    die("*** Bad data in destx: $destx.\n");
89 90 91 92 93
}
if ($desty =~ /^([\d.]+)$/) {
    $desty = $1;
}
else {
94
    die("*** Bad data in desty: $desty.\n");
95
}
96
if ($desto =~ /^([-\d.]+)$/) {
97 98 99
    $desto = $1;
}
else {
100
    die("*** Bad data in desto: $desto.\n");
101 102 103 104 105
}

#
# Check args and permission
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
106 107 108 109 110 111 112 113 114
#
my $this_user = User->ThisUser();
if (! defined($this_user)) {
    die("*** $0:\n".
	"    You ($UID) do not exist!\n");
}

my $node = Node->Lookup($nodeid);
if (!defined($node)) {
115 116 117
    die("*** $0:\n".
	"    Node does not exist: $nodeid\n");
}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
118
if (!$node->AccessCheck($this_user, TB_NODEACCESS_MODIFYINFO())) {
119 120 121 122
    die("*** $0:\n".
	"    Not enough permission: $nodeid\n");

}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
123 124
my $experiment = $node->Reservation();
if (!defined($experiment)) {
125 126 127
    die("*** $0:\n".
	"    Could not map nodeid to experiment: $nodeid\n");
}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
128 129 130
my $pid = $experiment->pid();
my $eid = $experiment->eid();
my $vname = $node->vname();
131 132 133 134 135 136 137 138

#
# Inject an event. 
#
my $inject_string = "$TEVC -e $pid/$eid now $vname SETDEST " .
    "X=" . $destx . " Y=" . $desty . " ORIENTATION=" . $desto;

if ($debug) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
139
    print $inject_string . "\n";
140 141 142 143 144 145 146 147
}

system($inject_string) &&
    die("*** $0:\n".
	"    Failed to inject setdest event!\n");
    
exit(0);