register_sliver.in 2.03 KB
Newer Older
1 2 3
#!/usr/bin/perl -w
#
# GENIPUBLIC-COPYRIGHT
Leigh B Stoller's avatar
Leigh B Stoller committed
4
# Copyright (c) 2008-2010 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 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
# All rights reserved.
#
use strict;
use English;
use Getopt::Std;

#
# Register slivers at their SAs
# 
sub usage()
{
    print "Usage: register_sliver [-d] [-u] slice_idx\n";
    exit(1);
}
my $optlist    = "ndu";
my $debug      = 1;
my $impotent   = 0;
my $unregister = 0;
sub fatal($);

#
# Configure variables
#
my $TB		  = "@prefix@";
my $TBOPS         = "@TBOPSEMAIL@";
my $TBLOGS        = "@TBLOGSEMAIL@";

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

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

#
# Check args.
#
my %options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (defined($options{"d"})) {
    $debug = 1;
}
if (defined($options{"n"})) {
    $impotent = 1;
}
if (defined($options{"u"})) {
    $unregister  = 1;
}
usage()
    if (@ARGV != 1);

use vars qw($GENI_DBNAME);
$GENI_DBNAME = "geni-cm";

# Now we can load the libraries after setting the proper DB.
use lib '@prefix@/lib';
use libtestbed;
require GeniDB;
require Genixmlrpc;
require GeniCertificate;
require GeniSlice;
require GeniCM;

my $slice = GeniSlice->Lookup($ARGV[0]);
if (!defined($slice)) {
    fatal("Could not find slice in the DB");
}

#
# The RPC context for this script is as the SA
#
my $EMULAB_PEMFILE = "@prefix@/etc/genicm.pem";
my $certificate  = GeniCertificate->LoadFromFile($EMULAB_PEMFILE);
if (!defined($certificate)) {
    fatal("Could not load certificate from $EMULAB_PEMFILE");
}
Leigh B Stoller's avatar
Leigh B Stoller committed
85
$ENV{'MYURN'} = "urn:publicid:IDN+@OURDOMAIN@+authority+cm";
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
Genixmlrpc->SetContext(Genixmlrpc->Context($certificate));
Genixmlrpc->SetTimeout(10);

if ($slice->Lock() != 0) {
    print STDERR "Could not lock slice $slice. Try again later.\n";
    exit(0);
}
my $retval = ($unregister ?
	      GeniCM::UnRegisterSliver($slice) :
	      GeniCM::RegisterSliver($slice));
$slice->UnLock();
exit($retval);

sub fatal($) {
    my ($msg) = @_;

    print STDERR "$msg\n";
    exit(1);
}