chiark / gitweb /
TODO
[autopkgtest.git] / virt-subproc / adt-virt-xenlvm
index 5cda2aab8799205f65ef626a4de5fe79237431e5..08608daba9ad6d6f321bb33b5b1f63dab365ecc3 100755 (executable)
@@ -3,7 +3,7 @@
 # adt-virt-xenlvm is part of autopkgtest
 # autopkgtest is a tool for testing Debian binary packages
 #
-# autopkgtest is Copyright (C) 2006 Canonical Ltd.
+# autopkgtest is Copyright (C) 2006-2007 Canonical Ltd.
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -26,6 +26,9 @@ import sys
 import os
 import re as regexp
 import string
+import subprocess
+import fnmatch
+import signal
 from optparse import OptionParser
 
 try: our_base = os.environ['AUTOPKGTEST_BASE']
@@ -34,32 +37,81 @@ sys.path.insert(1, our_base+'/python')
 
 import VirtSubproc as vsp
 
-witholder = None
+withholder = None
 downtmp = '/root/adt-downtmp'
+pauses = None
+event_counters = { }
+
+def check_pause(kind):
+       if not event_counters.has_key(kind): event_counters[kind] = 0
+       event = '%s#%d' % (kind, event_counters[kind])
+       event_counters[kind] += 1
+       for pattern in pauses:
+               if not '#' in pattern: matched = kind == pattern
+               else: matched = fnmatch.fnmatchcase(event,pattern)
+               if matched: break
+       if not matched:
+               return
+       print >>sys.stderr, ("----- adt-virt-xenlvm pause (event=%s),"
+                               " stopping -----" % event)
+       os.kill(0, signal.SIGSTOP)
 
 def parse_args():
-       global debuglevel, xlargs, gain_root
+       global debuglevel, with_testbed, console, pauses
 
        usage = "%prog <options> [-- <adt-xenlvm options>]"
        parser = OptionParser(usage=usage)
        pa = parser.add_option
        pe = parser.error
 
-       pa('-r', '--gain-root', type='string', dest='gain_root');
-       pa('-d', '--debug', action='store_true', dest='debug');
+       pa('-r', '--gain-root', type='string', dest='gain_root')
+       pa('-d', '--debug', action='store_true', dest='debug')
+       pa('', '--userv', action='store_true', dest='userv')
+       pa('', '--distro', type='string', dest='distro')
+       pa('', '--nominum', type='string', dest='nominum')
+       pa('','--pause', type='string', dest='pause', default='')
 
        (opts,xlargs) = parser.parse_args()
        vsp.debuglevel = opts.debug
-
-       if opts.gain_root = None: gain_root = []
-       else: gain_root = opts.gain_root.split()
-       vsp.down = gain_root + ['adt-xenlvm-on-testbed'] + xlargs + ['--']
+       xargs_userv = []
+       xargs_direct = []
+
+       for k in ['distro','nominum']:
+               v = getattr(opts,k)
+               if v is None: continue
+               xargs_direct.append('--%s=%s' % (k, v))
+               xargs_userv.append('-D%s=%s' % (k, v))
+
+       if not opts.userv:
+               if opts.gain_root is None: gain_root = []
+               else: gain_root = opts.gain_root.split()
+               with_testbed = (gain_root + ['adt-xenlvm-with-testbed'] +
+                       xargs_direct + xlargs +
+                       ['--','sh','-ec','echo y; exec cat'])
+               vsp.down = (gain_root + ['adt-xenlvm-on-testbed'] +
+                       xargs_direct + xlargs + ['--'])
+       else:
+               if opts.gain_root:
+                       pe('--userv and --gain-root are not compatible')
+               basis = (['userv'] + xargs_userv + xlargs +
+                       ['root','adt-xenlvm-testbed'])
+               with_testbed = basis + ['with']
+               get_down = subprocess.Popen(basis + ['pon0'],
+                       stdin=file('/dev/null'), stdout=subprocess.PIPE,
+                       stderr=None)
+               (pon0, _) = get_down.communicate()
+               if get_down.returncode:
+                       vsp.bomb('failed to check userv service provision'
+                               ' and subcommand details (code=%d)' %
+                               get_down.returncode)
+               vsp.down = pon0.split('\0')
+       pauses = opts.pause.split(',')
 
 def do_open():
+       global withholder
        assert(withholder is None)
        withholder = subprocess.Popen(
-               gain_root + ['adt-xenlvm-with-testbed'] + xlargs +
-                ['--','sh','-ec','echo y; exec cat'],
+               with_testbed,
                stdin=subprocess.PIPE, stdout=subprocess.PIPE )
        l = withholder.stdout.readline(2)
        rc = withholder.poll()
@@ -67,34 +119,45 @@ def do_open():
                withholder.stdin.close()
                withholder.stdout.close()
                withholder = None
-               bomb("with-testbed failed, code %d" % rc)
+               vsp.bomb("with-testbed failed, code %d" % rc)
        if l != "y\n":
-               bomb("with-testbed sh gave wrong output `%s', not `l'"
+               vsp.bomb("with-testbed sh gave wrong output `%s', not `l'"
                        % l.rstrip("\n"))
        vsp.execute('mkdir %s' % downtmp, downp=True)   
 
 def do_close():
+       global withholder
+       check_pause('close')
        withholder.stdin.close()
        withholder.stdout.close()
        rc = withholder.wait()
        withholder = None
-       if rc: bomb("with-testbed failed when closing/resetting, code %d" % rc)
+       if rc: vsp.bomb("with-testbed failed when closing/reverting,"
+                       " code %d" % rc)
+
+def hook_forked_inchild():
+       if withholder is not None:
+               withholder.stdin.close()
+               withholder.stdout.close()
 
 def hook_open():
        hook_cleanup()
        do_open()
        return downtmp
 
-def hook_reset():
+def hook_revert():
+       check_pause('revert')
        do_close()
        do_open()
 
 def hook_cleanup():
+       check_pause('cleanup')
+       global withholder
        if withholder is not None:
                do_close()
 
 def hook_capabilities():
-       return ('revert','root-on-testbed','suggest-normal-user=adtxenu')
+       return ['revert','root-on-testbed','suggested-normal-user=adtxenu']
 
 parse_args()
 vsp.main()