virt-subproc/adt-virt-chroot

   1 #!/usr/bin/python2.4
   2 #
   3 # adt-virt-chroot is part of autopkgtest
   4 # autopkgtest is a tool for testing Debian binary packages
   5 #
   6 # autopkgtest is Copyright (C) 2006-2007 Canonical Ltd.
   7 #
   8 # This program is free software; you can redistribute it and/or modify
   9 # it under the terms of the GNU General Public License as published by
  10 # the Free Software Foundation; either version 2 of the License, or
  11 # (at your option) any later version.
  12 #
  13 # This program is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 # GNU General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program; if not, write to the Free Software
  20 # Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 #
  22 # See the file CREDITS for a full list of credits information (often
  23 # installed as /usr/share/doc/autopkgtest/CREDITS).
  24
  25 import sys
  26 import os
  27 import string
  28 from optparse import OptionParser
  29
  30 try: our_base = os.environ['AUTOPKGTEST_BASE']
  31 except KeyError: our_base = '/usr/share/autopkgtest';
  32 sys.path.insert(1, our_base+'/python')
  33
  34 import VirtSubproc as vsp
  35 capabilities = []
  36
  37 def parse_args():
  38         global down, debuglevel
  39
  40         usage = "%prog [<options>] =<dchroot>|/path/to/chroot"
  41         parser = OptionParser(usage=usage)
  42         pa = parser.add_option
  43         pe = parser.error
  44
  45         pa('-r', '--gain-root', type='string', dest='gain_root');
  46         pa('-d', '--debug', action='store_true', dest='debug');
  47
  48         (opts,args) = parser.parse_args()
  49         if len(args) != 1: pe("need exactly one arg, chroot specification")
  50
  51         vsp.debuglevel = opts.debug
  52
  53         chroot_arg = args[0]
  54         if not chroot_arg: pe("chroot specification may not be empty")
  55         if chroot_arg == '=': down = ['dchroot','-q']
  56         elif chroot_arg == '=': down = ['dchroot','-q']
  57         elif chroot_arg[0] == '=': down = ['dchroot','-q','-c',chroot_arg[1:]]
  58         elif chroot_arg[0] == '/': down = ['chroot',chroot_arg,'--']
  59         else: pe("chroot spec must be =[DCHROOT] or /PATH/TO/CHROOT")
  60
  61         if opts.gain_root != None:
  62                 down = opts.gain_root.split() + down
  63
  64         if opts.gain_root or os.getuid()==0:
  65                 capabilities.append('root-on-testbed')
  66
  67         vsp.down = down
  68
  69 def hook_open():
  70         global downtmp
  71         vsp.execute('true', downp=True)
  72         return None
  73
  74 def hook_cleanup():
  75         pass
  76
  77 def hook_forked_inchild():
  78         pass
  79
  80 def hook_capabilities():
  81         return capabilities
  82
  83 parse_args()
  84 vsp.main()