system/TestPlc.py

   1 # $Id$
   2 import os, os.path
   3 import datetime
   4 import time
   5 import sys
   6 import traceback
   7 from types import StringTypes
   8 import socket
   9
  10 import utils
  11 from TestSite import TestSite
  12 from TestNode import TestNode
  13 from TestUser import TestUser
  14 from TestKey import TestKey
  15 from TestSlice import TestSlice
  16 from TestSliver import TestSliver
  17 from TestBox import TestBox
  18 from TestSsh import TestSsh
  19 from TestApiserver import TestApiserver
  20
  21 # step methods must take (self) and return a boolean (options is a member of the class)
  22
  23 def standby(minutes,dry_run):
  24     utils.header('Entering StandBy for %d mn'%minutes)
  25     if dry_run:
  26         print 'dry_run'
  27     else:
  28         time.sleep(60*minutes)
  29     return True
  30
  31 def standby_generic (func):
  32     def actual(self):
  33         minutes=int(func.__name__.split("_")[1])
  34         return standby(minutes,self.options.dry_run)
  35     return actual
  36
  37 def node_mapper (method):
  38     def actual(self):
  39         overall=True
  40         node_method = TestNode.__dict__[method.__name__]
  41         for site_spec in self.plc_spec['sites']:
  42             test_site = TestSite (self,site_spec)
  43             for node_spec in site_spec['nodes']:
  44                 test_node = TestNode (self,test_site,node_spec)
  45                 if not node_method(test_node): overall=False
  46         return overall
  47     return actual
  48
  49 def slice_mapper_options (method):
  50     def actual(self):
  51         overall=True
  52         slice_method = TestSlice.__dict__[method.__name__]
  53         for slice_spec in self.plc_spec['slices']:
  54             site_spec = self.locate_site (slice_spec['sitename'])
  55             test_site = TestSite(self,site_spec)
  56             test_slice=TestSlice(self,test_site,slice_spec)
  57             if not slice_method(test_slice,self.options): overall=False
  58         return overall
  59     return actual
  60
  61 SEP='<sep>'
  62
  63 class TestPlc:
  64
  65     default_steps = ['uninstall','install','install_rpm',
  66                      'configure', 'start', SEP,
  67                      'store_keys', 'clear_known_hosts', 'initscripts', SEP,
  68                      'sites', 'nodes', 'slices', 'nodegroups', SEP,
  69                      'init_node','bootcd', 'configure_qemu', 'export_qemu',
  70                      'kill_all_qemus', 'reinstall_node','start_node', SEP,
  71                      'nodes_booted', 'nodes_ssh', 'check_slice',
  72                      'check_initscripts', 'check_tcp',SEP,
  73                      'force_gather_logs', 'force_kill_qemus', 'force_record_tracker','force_free_tracker' ]
  74     other_steps = [ 'stop_all_vservers','fresh_install', 'cache_rpm', 'stop', SEP,
  75                     'clean_initscripts', 'clean_sites', 'clean_nodes',
  76                     'clean_slices', 'clean_keys', SEP,
  77                     'show_boxes', 'list_all_qemus', 'list_qemus', SEP,
  78                     'db_dump' , 'db_restore', ' cleanup_tracker',
  79                     'standby_1 through 20'
  80                     ]
  81
  82     @staticmethod
  83     def printable_steps (list):
  84         return " ".join(list).replace(" "+SEP+" "," \\\n")
  85     @staticmethod
  86     def valid_step (step):
  87         return step != SEP
  88
  89     def __init__ (self,plc_spec,options):
  90         self.plc_spec=plc_spec
  91         self.options=options
  92         self.test_ssh=TestSsh(self.plc_spec['hostname'],self.options.buildname)
  93         try:
  94             self.vserverip=plc_spec['vserverip']
  95             self.vservername=plc_spec['vservername']
  96             self.url="https://%s:443/PLCAPI/"%plc_spec['vserverip']
  97             self.vserver=True
  98         except:
  99             raise Exception,'chroot-based myplc testing is deprecated'
 100         self.apiserver=TestApiserver(self.url,options.dry_run)
 101
 102     def name(self):
 103         name=self.plc_spec['name']
 104         return "%s.%s"%(name,self.vservername)
 105
 106     def hostname(self):
 107         return self.plc_spec['hostname']
 108
 109     def is_local (self):
 110         return self.test_ssh.is_local()
 111
 112     # define the API methods on this object through xmlrpc
 113     # would help, but not strictly necessary
 114     def connect (self):
 115         pass
 116
 117     def actual_command_in_guest (self,command):
 118         return self.test_ssh.actual_command(self.host_to_guest(command))
 119
 120     def run_in_guest (self,command):
 121         return utils.system(self.actual_command_in_guest(command))
 122
 123     def run_in_host (self,command):
 124         return self.test_ssh.run_in_buildname(command)
 125
 126     #command gets run in the vserver
 127     def host_to_guest(self,command):
 128         return "vserver %s exec %s"%(self.vservername,command)
 129
 130     # xxx quick n dirty
 131     def run_in_guest_piped (self,local,remote):
 132         return utils.system(local+" | "+self.test_ssh.actual_command(self.host_to_guest(remote),keep_stdin=True))
 133
 134     def auth_root (self):
 135         return {'Username':self.plc_spec['PLC_ROOT_USER'],
 136                 'AuthMethod':'password',
 137                 'AuthString':self.plc_spec['PLC_ROOT_PASSWORD'],
 138                 'Role' : self.plc_spec['role']
 139                 }
 140     def locate_site (self,sitename):
 141         for site in self.plc_spec['sites']:
 142             if site['site_fields']['name'] == sitename:
 143                 return site
 144             if site['site_fields']['login_base'] == sitename:
 145                 return site
 146         raise Exception,"Cannot locate site %s"%sitename
 147
 148     def locate_node (self,nodename):
 149         for site in self.plc_spec['sites']:
 150             for node in site['nodes']:
 151                 if node['name'] == nodename:
 152                     return (site,node)
 153         raise Exception,"Cannot locate node %s"%nodename
 154
 155     def locate_hostname (self,hostname):
 156         for site in self.plc_spec['sites']:
 157             for node in site['nodes']:
 158                 if node['node_fields']['hostname'] == hostname:
 159                     return (site,node)
 160         raise Exception,"Cannot locate hostname %s"%hostname
 161
 162     def locate_key (self,keyname):
 163         for key in self.plc_spec['keys']:
 164             if key['name'] == keyname:
 165                 return key
 166         raise Exception,"Cannot locate key %s"%keyname
 167
 168     def locate_slice (self, slicename):
 169         for slice in self.plc_spec['slices']:
 170             if slice['slice_fields']['name'] == slicename:
 171                 return slice
 172         raise Exception,"Cannot locate slice %s"%slicename
 173
 174     # all different hostboxes used in this plc
 175     def gather_hostBoxes(self):
 176         # maps on sites and nodes, return [ (host_box,test_node) ]
 177         tuples=[]
 178         for site_spec in self.plc_spec['sites']:
 179             test_site = TestSite (self,site_spec)
 180             for node_spec in site_spec['nodes']:
 181                 test_node = TestNode (self, test_site, node_spec)
 182                 if not test_node.is_real():
 183                     tuples.append( (test_node.host_box(),test_node) )
 184         # transform into a dict { 'host_box' -> [ test_node .. ] }
 185         result = {}
 186         for (box,node) in tuples:
 187             if not result.has_key(box):
 188                 result[box]=[node]
 189             else:
 190                 result[box].append(node)
 191         return result
 192
 193     # a step for checking this stuff
 194     def show_boxes (self):
 195         for (box,nodes) in self.gather_hostBoxes().iteritems():
 196             print box,":"," + ".join( [ node.name() for node in nodes ] )
 197         return True
 198
 199     # make this a valid step
 200     def kill_all_qemus(self):
 201         # this is the brute force version, kill all qemus on that host box
 202         for (box,nodes) in self.gather_hostBoxes().iteritems():
 203             # pass the first nodename, as we don't push template-qemu on testboxes
 204             nodedir=nodes[0].nodedir()
 205             TestBox(box,self.options.buildname).kill_all_qemus(nodedir)
 206         return True
 207
 208     # make this a valid step
 209     def list_all_qemus(self):
 210         for (box,nodes) in self.gather_hostBoxes().iteritems():
 211             # this is the brute force version, kill all qemus on that host box
 212             TestBox(box,self.options.buildname).list_all_qemus()
 213         return True
 214
 215     # kill only the right qemus
 216     def list_qemus(self):
 217         for (box,nodes) in self.gather_hostBoxes().iteritems():
 218             # the fine-grain version
 219             for node in nodes:
 220                 node.list_qemu()
 221         return True
 222
 223     # kill only the right qemus
 224     def kill_qemus(self):
 225         for (box,nodes) in self.gather_hostBoxes().iteritems():
 226             # the fine-grain version
 227             for node in nodes:
 228                 node.kill_qemu()
 229         return True
 230
 231
 232     ### utility methods for handling the pool of IP addresses allocated to plcs
 233     # Logic
 234     # (*) running plcs are recorded in the file named ~/running-test-plcs
 235     # (*) this file contains a line for each running plc, older first
 236     # (*) each line contains the vserver name + the hostname of the (vserver) testbox where it sits
 237     # (*) the free_tracker method performs a vserver stop on the oldest entry
 238     # (*) the record_tracker method adds an entry at the bottom of the file
 239     # (*) the cleanup_tracker method stops all known vservers and removes the tracker file
 240
 241     TRACKER_FILE="~/running-test-plcs"
 242
 243     def record_tracker (self):
 244         command="echo %s %s >> %s"%(self.vservername,self.test_ssh.hostname,TestPlc.TRACKER_FILE)
 245         (code,output) = utils.output_of (self.test_ssh.actual_command(command))
 246         if code != 0:
 247             print "WARNING : COULD NOT record_tracker %s as a running plc on %s"%(self.vservername,self.test_ssh.hostname)
 248             return False
 249         print "Recorded %s in running plcs on host %s"%(self.vservername,self.test_ssh.hostname)
 250         return True
 251
 252     def free_tracker (self):
 253         command="head -1 %s"%TestPlc.TRACKER_FILE
 254         (code,line) = utils.output_of(self.test_ssh.actual_command(command))
 255         if code != 0:
 256             print "No entry found in %s on %s"%(TestPlc.TRACKER_FILE,self.test_ssh.hostname)
 257             return False
 258         try:
 259             [vserver_to_stop,hostname] = line.split()
 260         except:
 261             print "WARNING: free_tracker: Could not parse %s - skipped"%TestPlc.TRACKER_FILE
 262             return False
 263         stop_command = "vserver --silent %s stop"%vserver_to_stop
 264         utils.system(self.test_ssh.actual_command(stop_command))
 265         x=TestPlc.TRACKER_FILE
 266         flush_command = "tail --lines=+2 %s > %s.tmp ; mv %s.tmp %s"%(x,x,x,x)
 267         utils.system(self.test_ssh.actual_command(flush_command))
 268         return True
 269
 270     # this should/could stop only the ones in TRACKER_FILE if that turns out to be reliable
 271     def cleanup_tracker (self):
 272         stop_all = "cd /vservers ; for i in * ; do vserver --silent $i stop ; done"
 273         utils.system(self.test_ssh.actual_command(stop_all))
 274         clean_tracker = "rm -f %s"%TestPlc.TRACKER_FILE
 275         utils.system(self.test_ssh.actual_command(clean_tracker))
 276
 277     def uninstall(self):
 278         self.run_in_host("vserver --silent %s delete"%self.vservername)
 279         return True
 280
 281     ### install
 282     def install(self):
 283         # we need build dir for vtest-init-vserver
 284         if self.is_local():
 285             # a full path for the local calls
 286             build_dir=os.path.dirname(sys.argv[0])+"/build"
 287         else:
 288             # use a standard name - will be relative to remote buildname
 289             build_dir="build"
 290         # run checkout in any case - would do an update if already exists
 291         build_checkout = "svn checkout %s %s"%(self.options.build_url,build_dir)
 292         if self.run_in_host(build_checkout) != 0:
 293             return False
 294         # the repo url is taken from arch-rpms-url
 295         # with the last step (i386.) removed
 296         repo_url = self.options.arch_rpms_url
 297         for level in [ 'arch' ]:
 298             repo_url = os.path.dirname(repo_url)
 299         if self.options.arch == "i386":
 300             personality_option="-p linux32"
 301         else:
 302             personality_option="-p linux64"
 303         script="vtest-init-vserver.sh"
 304         vserver_name = self.vservername
 305         vserver_options="--netdev eth0 --interface %s"%self.vserverip
 306         try:
 307             vserver_hostname=socket.gethostbyaddr(self.vserverip)[0]
 308             vserver_options += " --hostname %s"%vserver_hostname
 309         except:
 310             pass
 311         create_vserver="%(build_dir)s/%(script)s %(personality_option)s %(vserver_name)s %(repo_url)s -- %(vserver_options)s"%locals()
 312         return self.run_in_host(create_vserver) == 0
 313
 314     ### install_rpm
 315     def install_rpm(self):
 316         return self.run_in_guest("yum -y install myplc-native")==0
 317
 318     ###
 319     def configure(self):
 320         tmpname='%s.plc-config-tty'%(self.name())
 321         fileconf=open(tmpname,'w')
 322         for var in [ 'PLC_NAME',
 323                      'PLC_ROOT_PASSWORD',
 324                      'PLC_ROOT_USER',
 325                      'PLC_MAIL_ENABLED',
 326                      'PLC_MAIL_SUPPORT_ADDRESS',
 327                      'PLC_DB_HOST',
 328                      'PLC_API_HOST',
 329                      'PLC_WWW_HOST',
 330                      'PLC_BOOT_HOST',
 331                      'PLC_NET_DNS1',
 332                      'PLC_NET_DNS2']:
 333             fileconf.write ('e %s\n%s\n'%(var,self.plc_spec[var]))
 334         fileconf.write('w\n')
 335         fileconf.write('q\n')
 336         fileconf.close()
 337         utils.system('cat %s'%tmpname)
 338         self.run_in_guest_piped('cat %s'%tmpname,'plc-config-tty')
 339         utils.system('rm %s'%tmpname)
 340         return True
 341
 342     def start(self):
 343         self.run_in_guest('service plc start')
 344         return True
 345
 346     def stop(self):
 347         self.run_in_guest('service plc stop')
 348         return True
 349
 350     # could use a TestKey class
 351     def store_keys(self):
 352         for key_spec in self.plc_spec['keys']:
 353                 TestKey(self,key_spec).store_key()
 354         return True
 355
 356     def clean_keys(self):
 357         utils.system("rm -rf %s/keys/"%os.path(sys.argv[0]))
 358
 359     def sites (self):
 360         return self.do_sites()
 361
 362     def clean_sites (self):
 363         return self.do_sites(action="delete")
 364
 365     def do_sites (self,action="add"):
 366         for site_spec in self.plc_spec['sites']:
 367             test_site = TestSite (self,site_spec)
 368             if (action != "add"):
 369                 utils.header("Deleting site %s in %s"%(test_site.name(),self.name()))
 370                 test_site.delete_site()
 371                 # deleted with the site
 372                 #test_site.delete_users()
 373                 continue
 374             else:
 375                 utils.header("Creating site %s & users in %s"%(test_site.name(),self.name()))
 376                 test_site.create_site()
 377                 test_site.create_users()
 378         return True
 379
 380     def nodes (self):
 381         return self.do_nodes()
 382     def clean_nodes (self):
 383         return self.do_nodes(action="delete")
 384
 385     def do_nodes (self,action="add"):
 386         for site_spec in self.plc_spec['sites']:
 387             test_site = TestSite (self,site_spec)
 388             if action != "add":
 389                 utils.header("Deleting nodes in site %s"%test_site.name())
 390                 for node_spec in site_spec['nodes']:
 391                     test_node=TestNode(self,test_site,node_spec)
 392                     utils.header("Deleting %s"%test_node.name())
 393                     test_node.delete_node()
 394             else:
 395                 utils.header("Creating nodes for site %s in %s"%(test_site.name(),self.name()))
 396                 for node_spec in site_spec['nodes']:
 397                     utils.pprint('Creating node %s'%node_spec,node_spec)
 398                     test_node = TestNode (self,test_site,node_spec)
 399                     test_node.create_node ()
 400         return True
 401
 402     # create nodegroups if needed, and populate
 403     # no need for a clean_nodegroups if we are careful enough
 404     def nodegroups (self):
 405         # 1st pass to scan contents
 406         groups_dict = {}
 407         for site_spec in self.plc_spec['sites']:
 408             test_site = TestSite (self,site_spec)
 409             for node_spec in site_spec['nodes']:
 410                 test_node=TestNode (self,test_site,node_spec)
 411                 if node_spec.has_key('nodegroups'):
 412                     nodegroupnames=node_spec['nodegroups']
 413                     if isinstance(nodegroupnames,StringTypes):
 414                         nodegroupnames = [ nodegroupnames ]
 415                     for nodegroupname in nodegroupnames:
 416                         if not groups_dict.has_key(nodegroupname):
 417                             groups_dict[nodegroupname]=[]
 418                         groups_dict[nodegroupname].append(test_node.name())
 419         auth=self.auth_root()
 420         for (nodegroupname,group_nodes) in groups_dict.iteritems():
 421             try:
 422                 self.apiserver.GetNodeGroups(auth,{'name':nodegroupname})[0]
 423             except:
 424                 self.apiserver.AddNodeGroup(auth,{'name':nodegroupname})
 425             for node in group_nodes:
 426                 self.apiserver.AddNodeToNodeGroup(auth,node,nodegroupname)
 427         return True
 428
 429     def all_hostnames (self) :
 430         hostnames = []
 431         for site_spec in self.plc_spec['sites']:
 432             hostnames += [ node_spec['node_fields']['hostname'] \
 433                            for node_spec in site_spec['nodes'] ]
 434         return hostnames
 435
 436     # gracetime : during the first <gracetime> minutes nothing gets printed
 437     def do_nodes_booted (self, minutes, gracetime,period=30):
 438         if self.options.dry_run:
 439             print 'dry_run'
 440             return True
 441         # compute timeout
 442         timeout = datetime.datetime.now()+datetime.timedelta(minutes=minutes)
 443         graceout = datetime.datetime.now()+datetime.timedelta(minutes=gracetime)
 444         # the nodes that haven't checked yet - start with a full list and shrink over time
 445         tocheck = self.all_hostnames()
 446         utils.header("checking nodes %r"%tocheck)
 447         # create a dict hostname -> status
 448         status = dict ( [ (hostname,'undef') for hostname in tocheck ] )
 449         while tocheck:
 450             # get their status
 451             tocheck_status=self.apiserver.GetNodes(self.auth_root(), tocheck, ['hostname','boot_state' ] )
 452             # update status
 453             for array in tocheck_status:
 454                 hostname=array['hostname']
 455                 boot_state=array['boot_state']
 456                 if boot_state == 'boot':
 457                     utils.header ("%s has reached the 'boot' state"%hostname)
 458                 else:
 459                     # if it's a real node, never mind
 460                     (site_spec,node_spec)=self.locate_hostname(hostname)
 461                     if TestNode.is_real_model(node_spec['node_fields']['model']):
 462                         utils.header("WARNING - Real node %s in %s - ignored"%(hostname,boot_state))
 463                         # let's cheat
 464                         boot_state = 'boot'
 465                     elif datetime.datetime.now() > graceout:
 466                         utils.header ("%s still in '%s' state"%(hostname,boot_state))
 467                         graceout=datetime.datetime.now()+datetime.timedelta(1)
 468                 status[hostname] = boot_state
 469             # refresh tocheck
 470             tocheck = [ hostname for (hostname,boot_state) in status.iteritems() if boot_state != 'boot' ]
 471             if not tocheck:
 472                 return True
 473             if datetime.datetime.now() > timeout:
 474                 for hostname in tocheck:
 475                     utils.header("FAILURE due to %s in '%s' state"%(hostname,status[hostname]))
 476                 return False
 477             # otherwise, sleep for a while
 478             time.sleep(period)
 479         # only useful in empty plcs
 480         return True
 481
 482     def nodes_booted(self):
 483         return self.do_nodes_booted(minutes=20,gracetime=15)
 484
 485     def do_nodes_ssh(self,minutes,gracetime,period=30):
 486         # compute timeout
 487         timeout = datetime.datetime.now()+datetime.timedelta(minutes=minutes)
 488         graceout = datetime.datetime.now()+datetime.timedelta(minutes=gracetime)
 489         tocheck = self.all_hostnames()
 490 #        self.scan_publicKeys(tocheck)
 491         utils.header("checking Connectivity on nodes %r"%tocheck)
 492         while tocheck:
 493             for hostname in tocheck:
 494                 # try to ssh in nodes
 495                 node_test_ssh = TestSsh (hostname,key="/etc/planetlab/root_ssh_key.rsa")
 496                 success=self.run_in_guest(node_test_ssh.actual_command("hostname"))==0
 497                 if success:
 498                     utils.header('The node %s is sshable -->'%hostname)
 499                     # refresh tocheck
 500                     tocheck.remove(hostname)
 501                 else:
 502                     # we will have tried real nodes once, in case they're up - but if not, just skip
 503                     (site_spec,node_spec)=self.locate_hostname(hostname)
 504                     if TestNode.is_real_model(node_spec['node_fields']['model']):
 505                         utils.header ("WARNING : check ssh access into real node %s - skipped"%hostname)
 506                         tocheck.remove(hostname)
 507                     elif datetime.datetime.now() > graceout:
 508                         utils.header("Could not ssh-enter root context on %s"%hostname)
 509             if  not tocheck:
 510                 return True
 511             if datetime.datetime.now() > timeout:
 512                 for hostname in tocheck:
 513                     utils.header("FAILURE to ssh into %s"%hostname)
 514                 return False
 515             # otherwise, sleep for a while
 516             time.sleep(period)
 517         # only useful in empty plcs
 518         return True
 519
 520     def nodes_ssh(self):
 521         return self.do_nodes_ssh(minutes=6,gracetime=4)
 522
 523     @node_mapper
 524     def init_node (self): pass
 525     @node_mapper
 526     def bootcd (self): pass
 527     @node_mapper
 528     def configure_qemu (self): pass
 529     @node_mapper
 530     def reinstall_node (self): pass
 531     @node_mapper
 532     def export_qemu (self): pass
 533
 534     def do_check_initscripts(self):
 535         overall = True
 536         for slice_spec in self.plc_spec['slices']:
 537             if not slice_spec.has_key('initscriptname'):
 538                 continue
 539             initscript=slice_spec['initscriptname']
 540             for nodename in slice_spec['nodenames']:
 541                 (site,node) = self.locate_node (nodename)
 542                 # xxx - passing the wrong site - probably harmless
 543                 test_site = TestSite (self,site)
 544                 test_slice = TestSlice (self,test_site,slice_spec)
 545                 test_node = TestNode (self,test_site,node)
 546                 test_sliver = TestSliver (self, test_node, test_slice)
 547                 if not test_sliver.check_initscript(initscript):
 548                     overall = False
 549         return overall
 550
 551     def check_initscripts(self):
 552             return self.do_check_initscripts()
 553
 554     def initscripts (self):
 555         for initscript in self.plc_spec['initscripts']:
 556             utils.pprint('Adding Initscript in plc %s'%self.plc_spec['name'],initscript)
 557             self.apiserver.AddInitScript(self.auth_root(),initscript['initscript_fields'])
 558         return True
 559
 560     def clean_initscripts (self):
 561         for initscript in self.plc_spec['initscripts']:
 562             initscript_name = initscript['initscript_fields']['name']
 563             print('Attempting to delete %s in plc %s'%(initscript_name,self.plc_spec['name']))
 564             try:
 565                 self.apiserver.DeleteInitScript(self.auth_root(),initscript_name)
 566                 print initscript_name,'deleted'
 567             except:
 568                 print 'deletion went wrong - probably did not exist'
 569         return True
 570
 571     def slices (self):
 572         return self.do_slices()
 573
 574     def clean_slices (self):
 575         return self.do_slices("delete")
 576
 577     def do_slices (self,  action="add"):
 578         for slice in self.plc_spec['slices']:
 579             site_spec = self.locate_site (slice['sitename'])
 580             test_site = TestSite(self,site_spec)
 581             test_slice=TestSlice(self,test_site,slice)
 582             if action != "add":
 583                 utils.header("Deleting slices in site %s"%test_site.name())
 584                 test_slice.delete_slice()
 585             else:
 586                 utils.pprint("Creating slice",slice)
 587                 test_slice.create_slice()
 588                 utils.header('Created Slice %s'%slice['slice_fields']['name'])
 589         return True
 590
 591     @slice_mapper_options
 592     def check_slice(self): pass
 593
 594     @node_mapper
 595     def clear_known_hosts (self): pass
 596
 597     @node_mapper
 598     def start_node (self) : pass
 599
 600     def all_sliver_objs (self):
 601         result=[]
 602         for slice_spec in self.plc_spec['slices']:
 603             slicename = slice_spec['slice_fields']['name']
 604             for nodename in slice_spec['nodenames']:
 605                 result.append(self.locate_sliver_obj (nodename,slicename))
 606         return result
 607
 608     def locate_sliver_obj (self,nodename,slicename):
 609         (site,node) = self.locate_node(nodename)
 610         slice = self.locate_slice (slicename)
 611         # build objects
 612         test_site = TestSite (self, site)
 613         test_node = TestNode (self, test_site,node)
 614         # xxx the slice site is assumed to be the node site - mhh - probably harmless
 615         test_slice = TestSlice (self, test_site, slice)
 616         return TestSliver (self, test_node, test_slice)
 617
 618     def check_tcp (self):
 619         specs = self.plc_spec['tcp_test']
 620         overall=True
 621         for spec in specs:
 622             port = spec['port']
 623             # server side
 624             s_test_sliver = self.locate_sliver_obj (spec['server_node'],spec['server_slice'])
 625             if not s_test_sliver.run_tcp_server(port,timeout=10):
 626                 overall=False
 627                 break
 628
 629             # idem for the client side
 630             c_test_sliver = self.locate_sliver_obj(spec['server_node'],spec['server_slice'])
 631             if not c_test_sliver.run_tcp_client(s_test_sliver.test_node.name(),port):
 632                 overall=False
 633         return overall
 634
 635
 636     def gather_logs (self):
 637         # (1) get the plc's /var/log and store it locally in logs/myplc.var-log.<plcname>/*
 638         # (2) get all the nodes qemu log and store it as logs/node.qemu.<node>.log
 639         # (3) get the nodes /var/log and store is as logs/node.var-log.<node>/*
 640         # (4) as far as possible get the slice's /var/log as logs/sliver.var-log.<sliver>/*
 641         # (1)
 642         print "-------------------- TestPlc.gather_logs : PLC's /var/log"
 643         self.gather_var_logs ()
 644         # (2)
 645         print "-------------------- TestPlc.gather_logs : nodes's QEMU logs"
 646         for site_spec in self.plc_spec['sites']:
 647             test_site = TestSite (self,site_spec)
 648             for node_spec in site_spec['nodes']:
 649                 test_node=TestNode(self,test_site,node_spec)
 650                 test_node.gather_qemu_logs()
 651         # (3)
 652         print "-------------------- TestPlc.gather_logs : nodes's /var/log"
 653         self.gather_nodes_var_logs()
 654         # (4)
 655         print "-------------------- TestPlc.gather_logs : sample sliver's /var/log"
 656         self.gather_slivers_var_logs()
 657         return True
 658
 659     def gather_slivers_var_logs(self):
 660         for test_sliver in self.all_sliver_objs():
 661             remote = test_sliver.tar_var_logs()
 662             utils.system("mkdir -p logs/sliver.var-log.%s"%test_sliver.name())
 663             command = remote + " | tar -C logs/sliver.var-log.%s -xf -"%test_sliver.name()
 664             utils.system(command)
 665         return True
 666
 667     def gather_var_logs (self):
 668         to_plc = self.actual_command_in_guest("tar -C /var/log/ -cf - .")
 669         command = to_plc + "| tar -C logs/myplc.var-log.%s -xf -"%self.name()
 670         utils.system("mkdir -p logs/myplc.var-log.%s"%self.name())
 671         utils.system(command)
 672
 673     def gather_nodes_var_logs (self):
 674         for site_spec in self.plc_spec['sites']:
 675             test_site = TestSite (self,site_spec)
 676             for node_spec in site_spec['nodes']:
 677                 test_node=TestNode(self,test_site,node_spec)
 678                 test_ssh = TestSsh (test_node.name(),key="/etc/planetlab/root_ssh_key.rsa")
 679                 to_plc = self.actual_command_in_guest ( test_ssh.actual_command("tar -C /var/log -cf - ."))
 680                 command = to_plc + "| tar -C logs/node.var-log.%s -xf -"%test_node.name()
 681                 utils.system("mkdir -p logs/node.var-log.%s"%test_node.name())
 682                 utils.system(command)
 683
 684
 685     # returns the filename to use for sql dump/restore, using options.dbname if set
 686     def dbfile (self, database):
 687         # uses options.dbname if it is found
 688         try:
 689             name=self.options.dbname
 690             if not isinstance(name,StringTypes):
 691                 raise Exception
 692         except:
 693             t=datetime.datetime.now()
 694             d=t.date()
 695             name=str(d)
 696         return "/root/%s-%s.sql"%(database,name)
 697
 698     def db_dump(self):
 699         dump=self.dbfile("planetab4")
 700         self.run_in_guest('pg_dump -U pgsqluser planetlab4 -f '+ dump)
 701         utils.header('Dumped planetlab4 database in %s'%dump)
 702         return True
 703
 704     def db_restore(self):
 705         dump=self.dbfile("planetab4")
 706         ##stop httpd service
 707         self.run_in_guest('service httpd stop')
 708         # xxx - need another wrapper
 709         self.run_in_guest_piped('echo drop database planetlab4','psql --user=pgsqluser template1')
 710         self.run_in_guest('createdb -U postgres --encoding=UNICODE --owner=pgsqluser planetlab4')
 711         self.run_in_guest('psql -U pgsqluser planetlab4 -f '+dump)
 712         ##starting httpd service
 713         self.run_in_guest('service httpd start')
 714
 715         utils.header('Database restored from ' + dump)
 716
 717     @standby_generic
 718     def standby_1(): pass
 719     @standby_generic
 720     def standby_2(): pass
 721     @standby_generic
 722     def standby_3(): pass
 723     @standby_generic
 724     def standby_4(): pass
 725     @standby_generic
 726     def standby_5(): pass
 727     @standby_generic
 728     def standby_6(): pass
 729     @standby_generic
 730     def standby_7(): pass
 731     @standby_generic
 732     def standby_8(): pass
 733     @standby_generic
 734     def standby_9(): pass
 735     @standby_generic
 736     def standby_10(): pass
 737     @standby_generic
 738     def standby_11(): pass
 739     @standby_generic
 740     def standby_12(): pass
 741     @standby_generic
 742     def standby_13(): pass
 743     @standby_generic
 744     def standby_14(): pass
 745     @standby_generic
 746     def standby_15(): pass
 747     @standby_generic
 748     def standby_16(): pass
 749     @standby_generic
 750     def standby_17(): pass
 751     @standby_generic
 752     def standby_18(): pass
 753     @standby_generic
 754     def standby_19(): pass
 755     @standby_generic
 756     def standby_20(): pass
 757