Don't rely on exported $ROOT.
[invirt/packages/invirt-xen-config.git] / 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))