files/usr/sbin/sipb-xen-losetup

   1 #!/usr/bin/env python2.5
   2
   3 import sys
   4 import os
   5 from subprocess import call
   6
   7 def losetup(source, offset=0):
   8   # XXX we avoid colliding with other instances of ourself,
   9   #     but when it comes to other loop-device users we just
  10   #     pick a range things don't seem to use and hope...
  11   lockfilename = '/tmp/losetup.lock'
  12   os.close(os.open(lockfilename, os.O_CREAT+os.O_EXCL)) #lock
  13   try:
  14     loopdevice = None
  15     for i in xrange(32,60): # totally arbitrary, just looks to be unused on black-mesa
  16       filename = '/dev/loop%d'%i
  17       if 0 == len(file(filename).read(1)):
  18         loopdevice = filename # it's empty
  19         break
  20     if loopdevice is not None:
  21       call(['losetup', '-o', str(offset), loopdevice, source])
  22     else:
  23       raise RuntimeError('out of loop devices for copying VM image: too many at once?')
  24   finally:
  25     os.unlink(lockfilename) #unlock
  26   return loopdevice
  27
  28 def main(*argv):
  29   args = argv[1:]
  30   os.environ['PATH'] = '/usr/sbin:/usr/bin:/sbin:/bin'
  31   if not (1 <= len(args) <= 2):
  32     print >>sys.stderr, 'usage: %s sourcedevice [offset]' % argv[0]
  33     print >>sys.stderr, 'prints resulting loopback device; don\'t forget to losetup -d'
  34     return 2
  35   print losetup(*args)
  36   return 0
  37
  38 if __name__ == '__main__':
  39   sys.exit(main(*sys.argv))