# 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
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']
except KeyError: our_base = '/usr/share/autopkgtest';
sys.path.insert(1, our_base+'/python')
-import string
-from optparse import OptionParser
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 down, debuglevel, xlargs
+ global debuglevel, with_testbed, console, pauses
- usage = "%prog [<adt-xenlvm options>]"
- for a in
+ usage = "%prog <options> [-- <adt-xenlvm options>]"
+ parser = OptionParser(usage=usage)
+ pa = parser.add_option
+ pe = parser.error
- fixme must check args fit --<var>=<value> as expected by adt-xenlvm
- fixme then put set of args into global xlargs
- fixme fix these next two lines:
- vsp.debug("down = %s" % string.join(down))
- vsp.down = down
+ 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='')
-def hook_open():
- hook_cleanup()
+ (opts,xlargs) = parser.parse_args()
+ vsp.debuglevel = opts.debug
+ 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(
- ['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()
- if rc is not None: bomb("with-testbed failed, code %d" % rc)
- if l != "y\n": bomb("with-testbed sh gave wrong output `%s', not `l'"
+ if rc is not None:
+ withholder.stdin.close()
+ withholder.stdout.close()
+ withholder = None
+ vsp.bomb("with-testbed failed, code %d" % rc)
+ if l != "y\n":
+ vsp.bomb("with-testbed sh gave wrong output `%s', not `l'"
% l.rstrip("\n"))
- downtmp = '/root/adt-downtmp'
- vsp.execute('mkdir %s' % downtmp, downp=True)
- return downtmp
+ vsp.execute('mkdir %s' % downtmp, downp=True)
-def hook_stop():
+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 at shutdown, code %d" % rc)
+ if rc: vsp.bomb("with-testbed failed when closing/reverting,"
+ " code %d" % rc)
-def hook_cleanup():
+def hook_forked_inchild():
if withholder is not None:
withholder.stdin.close()
withholder.stdout.close()
- rc = withholder.wait()
- withholder = None
- if rc: print >>sys.stderr, (
- "with-testbed failed during cleanup, code %d" % rc)
-
+
+def hook_open():
+ hook_cleanup()
+ do_open()
+ return downtmp
+
+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','suggested-normal-user=adtxenu']
+
parse_args()
vsp.main()