mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
112 lines
2.8 KiB
Python
Executable File
112 lines
2.8 KiB
Python
Executable File
#!/usr/bin/python
|
|
#
|
|
# Copyright (c) Citrix Systems 2007-2011
|
|
# Author: Gianni Tedesco and Dave Scott
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published
|
|
# by the Free Software Foundation; version 2.1 only. with the special
|
|
# exception on linking described in file LICENSE.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
|
|
from xen.lowlevel import xs
|
|
|
|
def do_cloexec(fd):
|
|
from fcntl import fcntl, F_GETFD, F_SETFD, FD_CLOEXEC
|
|
|
|
ret = fcntl(fd, F_GETFD, FD_CLOEXEC)
|
|
if ret < 0:
|
|
return
|
|
fcntl(fd, F_SETFD, ret|FD_CLOEXEC)
|
|
|
|
def write_dm_pid(domid, pid):
|
|
store = xs.xs()
|
|
store.write('', '/local/domain/%d/qemu-pid'%domid, '%d'%pid)
|
|
|
|
def is_sdk():
|
|
try:
|
|
store = xs.xs(socket=True)
|
|
domid_as_dom0 = int(store.read('', 'domid'))
|
|
store.close()
|
|
assert domid_as_dom0 == 0
|
|
store = xs.xs(xenbus=True)
|
|
domid_on_host = int(store.read('', 'domid'))
|
|
store.close()
|
|
return domid_on_host != domid_as_dom0
|
|
except:
|
|
return False
|
|
|
|
def fake_dm(domid):
|
|
from time import sleep
|
|
from os import getpid
|
|
|
|
store = xs.xs()
|
|
|
|
store.write('', '/local/domain/%d/qemu-pid'%domid, '%d'%getpid())
|
|
store.write('', '/local/domain/%d/device-misc/dm-ready'%domid, '1')
|
|
store.write('', '/local/domain/%d/console/vnc-port'%domid, '%d'%(domid))
|
|
|
|
while True:
|
|
sleep(600)
|
|
|
|
return 0
|
|
|
|
def cleanup(domid):
|
|
from xen.lowlevel import xc
|
|
hcall = xc.xc()
|
|
|
|
print 'Unexpected termination, cleaning up...'
|
|
|
|
try:
|
|
hcall.domain_destroy(domid)
|
|
except xc.Error:
|
|
# we could have raced with another domain shutdown
|
|
pass
|
|
|
|
def enable_core_dumps():
|
|
from resource import getrlimit, RLIMIT_CORE, setrlimit
|
|
|
|
limit = 64 * 1024 * 1024
|
|
oldlimits = getrlimit(RLIMIT_CORE)
|
|
setrlimit(RLIMIT_CORE, (limit, oldlimits[1]))
|
|
return limit
|
|
|
|
def main(argv):
|
|
import os
|
|
|
|
qemu_env = os.environ
|
|
qemu_dm = '/usr/lib/xen/bin/qemu-dm'
|
|
domid = int(argv[1])
|
|
qemu_args = ['qemu-dm-%d'%domid] + argv[2:]
|
|
try:
|
|
vnc_idx = qemu_args.index("-vnc")
|
|
del qemu_args[vnc_idx: vnc_idx+2]
|
|
qemu_args.insert(vnc_idx, "0.0.0.0:1")
|
|
qemu_args.insert(vnc_idx, "-vnc")
|
|
except ValueError, IndexError:
|
|
pass
|
|
|
|
if is_sdk() is True:
|
|
return fake_dm(domid)
|
|
|
|
print "qemu-dm-wrapper in python:"
|
|
print "Using domid: %d" % domid
|
|
print "Arguments: %s" % " ".join(argv[1:])
|
|
print "everything else is from qemu-dm:"
|
|
|
|
core_dump_limit = enable_core_dumps()
|
|
print "core dump limit: %d" % core_dump_limit
|
|
|
|
write_dm_pid(domid, os.getpid())
|
|
|
|
os.dup2(1, 2)
|
|
os.execve(qemu_dm, qemu_args, qemu_env)
|
|
|
|
if __name__ == '__main__':
|
|
from sys import argv
|
|
raise SystemExit, main(argv)
|