# Author: acb@cs.princeton.edu/sapanb@cs.princeton.edu
import sys
+import os
from xmlrpclib import ServerProxy
-if (len(sys.argv)!=2):
- print "[FAILED] Please pass the name of the slice this script runs in as the first argument.\n";
-else:
- nodemanager = ServerProxy('http://127.0.0.1:812/')
- try:
- nodemanager.SetLoans(sys.argv[1], [])
- except:
- print "[FAILED] SetLoans didn't work, probably because the packets of this connection are not getting tagged.\n";
+homedir=os.environ['HOME']
+slice_name = homedir.rsplit('/')[0]
+
+nodemanager = ServerProxy('http://127.0.0.1:812/')
+try:
+ nodemanager.SetLoans(slice_name, [])
+except:
+ print "[FAILED] SetLoans didn't work, probably because the packets of this connection are not getting tagged.\n";
if (!-e "$ttraceroute") {
print $ttraceroute."\n";
- die("[FAILED] Please install tcptraceroute in the slice before running this test\n");
+ print "[OOPS] TCPtraceroute not found. Installing... \n";
+ system("yum -y tcptraceroute");
+ if (!-e "$ttraceroute") {
+ print "[FAILED] TCPtraceroute could not be installed. Test failed.\n";
+ }
}
else {
print "Found tcptraceroute. Good.\n";