Merge branch 'hvm' of 10.141.1.101:/var/lib/qubes/git/marmarek/core into hvm

This commit is contained in:
Joanna Rutkowska 2012-06-14 10:43:45 +02:00
commit 7465fc4182
3 changed files with 65 additions and 16 deletions

View File

@ -253,12 +253,13 @@ class QubesVm(object):
"kernel": { "default": None, 'order': 30 }, "kernel": { "default": None, 'order': 30 },
"uses_default_kernel": { "default": True, 'order': 30 }, "uses_default_kernel": { "default": True, 'order': 30 },
"uses_default_kernelopts": { "default": True, 'order': 30 }, "uses_default_kernelopts": { "default": True, 'order': 30 },
"kernelopts": { "default": "", 'order': 30, "eval": \ "kernelopts": { "default": "", 'order': 31, "eval": \
'value if not self.uses_default_kernelopts else default_kernelopts_pcidevs if len(self.pcidevs) > 0 else default_kernelopts' }, 'value if not self.uses_default_kernelopts else default_kernelopts_pcidevs if len(self.pcidevs) > 0 else default_kernelopts' },
"mac": { "attr": "_mac", "default": None }, "mac": { "attr": "_mac", "default": None },
"include_in_backups": { "default": True }, "include_in_backups": { "default": True },
"services": { "default": {}, "eval": "eval(str(value))" }, "services": { "default": {}, "eval": "eval(str(value))" },
"debug": { "default": False }, "debug": { "default": False },
"default_user": { "default": "user" },
##### Internal attributes - will be overriden in __init__ regardless of args ##### Internal attributes - will be overriden in __init__ regardless of args
"appmenus_templates_dir": { "eval": \ "appmenus_templates_dir": { "eval": \
'self.dir_path + "/" + default_appmenus_templates_subdir if self.updateable else ' + \ 'self.dir_path + "/" + default_appmenus_templates_subdir if self.updateable else ' + \
@ -268,6 +269,7 @@ class QubesVm(object):
"kernels_dir": { 'eval': 'qubes_kernels_base_dir + "/" + self.kernel if self.kernel is not None else ' + \ "kernels_dir": { 'eval': 'qubes_kernels_base_dir + "/" + self.kernel if self.kernel is not None else ' + \
# for backward compatibility (or another rare case): kernel=None -> kernel in VM dir # for backward compatibility (or another rare case): kernel=None -> kernel in VM dir
'self.dir_path + "/" + default_kernels_subdir' }, 'self.dir_path + "/" + default_kernels_subdir' },
"_start_guid_first": { 'eval': 'False' },
} }
### Mark attrs for XML inclusion ### Mark attrs for XML inclusion
@ -275,7 +277,8 @@ class QubesVm(object):
for prop in ['qid', 'name', 'dir_path', 'memory', 'maxmem', 'pcidevs', 'vcpus', 'internal',\ for prop in ['qid', 'name', 'dir_path', 'memory', 'maxmem', 'pcidevs', 'vcpus', 'internal',\
'uses_default_kernel', 'kernel', 'uses_default_kernelopts',\ 'uses_default_kernel', 'kernel', 'uses_default_kernelopts',\
'kernelopts', 'services', 'installed_by_rpm',\ 'kernelopts', 'services', 'installed_by_rpm',\
'uses_default_netvm', 'include_in_backups', 'debug' ]: 'uses_default_netvm', 'include_in_backups', 'debug',\
'default_user' ]:
attrs[prop]['save'] = 'str(self.%s)' % prop attrs[prop]['save'] = 'str(self.%s)' % prop
# Simple paths # Simple paths
for prop in ['conf_file', 'root_img', 'volatile_img', 'private_img']: for prop in ['conf_file', 'root_img', 'volatile_img', 'private_img']:
@ -650,12 +653,18 @@ class QubesVm(object):
return "NA" return "NA"
def is_fully_usable(self): def is_guid_running(self):
xid = self.get_xid() xid = self.get_xid()
if xid < 0: if xid < 0:
return False return False
if not os.path.exists('/var/run/qubes/guid_running.%d' % xid): if not os.path.exists('/var/run/qubes/guid_running.%d' % xid):
return False return False
return True
def is_fully_usable(self):
# Running gui-daemon implies also VM running
if not self.is_guid_running():
return False
# currently qrexec daemon doesn't cleanup socket in /var/run/qubes, so # currently qrexec daemon doesn't cleanup socket in /var/run/qubes, so
# it can be left from some other VM # it can be left from some other VM
return True return True
@ -1312,7 +1321,7 @@ class QubesVm(object):
raise QubesException("Not enough memory to start '{0}' VM! Close one or more running VMs and try again.".format(self.name)) raise QubesException("Not enough memory to start '{0}' VM! Close one or more running VMs and try again.".format(self.name))
xid = self.get_xid() xid = self.get_xid()
if os.getenv("DISPLAY") is not None and not os.path.isfile("/var/run/qubes/guid_running.{0}".format(xid)): if os.getenv("DISPLAY") is not None and not self.is_guid_running():
self.start_guid(verbose = verbose, notify_function = notify_function) self.start_guid(verbose = verbose, notify_function = notify_function)
args = [qrexec_client_path, "-d", str(xid), command] args = [qrexec_client_path, "-d", str(xid), command]
@ -1473,10 +1482,13 @@ class QubesVm(object):
# the successful unpause is some indicator of it # the successful unpause is some indicator of it
qmemman_client.close() qmemman_client.close()
if self._start_guid_first and start_guid and not preparing_dvm and os.path.exists('/var/run/shm.id'):
self.start_guid(verbose=verbose)
if not preparing_dvm: if not preparing_dvm:
self.start_qrexec_daemon(verbose=verbose) self.start_qrexec_daemon(verbose=verbose)
if start_guid and not preparing_dvm and os.path.exists('/var/run/shm.id'): if not self._start_guid_first and start_guid and not preparing_dvm and os.path.exists('/var/run/shm.id'):
self.start_guid(verbose=verbose) self.start_guid(verbose=verbose)
if preparing_dvm: if preparing_dvm:
@ -2221,6 +2233,8 @@ class QubesHVm(QubesVm):
attrs['drive'] = { 'save': 'str(self.drive)' } attrs['drive'] = { 'save': 'str(self.drive)' }
attrs['maxmem'].pop('save') attrs['maxmem'].pop('save')
attrs['timezone'] = { 'default': 'localtime', 'save': 'str(self.timezone)' } attrs['timezone'] = { 'default': 'localtime', 'save': 'str(self.timezone)' }
attrs['qrexec_installed'] = { 'default': False, 'save': 'str(self.qrexec_installed)' }
attrs['_start_guid_first']['eval'] = 'True'
return attrs return attrs
@ -2391,14 +2405,21 @@ class QubesHVm(QubesVm):
return "vif{0}.+".format(self.stubdom_xid) return "vif{0}.+".format(self.stubdom_xid)
def run(self, command, **kwargs): def run(self, command, **kwargs):
raise NotImplementedError("Needs qrexec agent - TODO") if self.qrexec_installed:
super(QubesHVm, self).run(command, **kwargs)
else:
raise QubesException("Needs qrexec agent installed in VM to use this function. See also qvm-prefs.")
@property @property
def stubdom_xid(self): def stubdom_xid(self):
if self.xid < 0: if self.xid < 0:
return -1 return -1
return int(xs.read('', '/local/domain/%d/image/device-model-domid' % self.xid)) stubdom_xid_str = xs.read('', '/local/domain/%d/image/device-model-domid' % self.xid)
if stubdom_xid_str is not None:
return int(stubdom_xid_str)
else:
return -1
def start_guid(self, verbose = True, notify_function = None): def start_guid(self, verbose = True, notify_function = None):
if verbose: if verbose:
@ -2409,7 +2430,8 @@ class QubesHVm(QubesVm):
raise QubesException("Cannot start qubes_guid!") raise QubesException("Cannot start qubes_guid!")
def start_qrexec_daemon(self, **kwargs): def start_qrexec_daemon(self, **kwargs):
pass if self.qrexec_installed:
super(QubesHVm, self).start_qrexec_daemon(**kwargs)
def pause(self): def pause(self):
if dry_run: if dry_run:
@ -2425,7 +2447,7 @@ class QubesHVm(QubesVm):
xc.domain_unpause(self.stubdom_xid) xc.domain_unpause(self.stubdom_xid)
super(QubesHVm, self).unpause() super(QubesHVm, self).unpause()
def is_fully_usable(self): def is_guid_running(self):
xid = self.stubdom_xid xid = self.stubdom_xid
if xid < 0: if xid < 0:
return False return False
@ -2795,7 +2817,7 @@ class QubesVmCollection(dict):
"installed_by_rpm", "internal", "installed_by_rpm", "internal",
"uses_default_netvm", "label", "memory", "vcpus", "pcidevs", "uses_default_netvm", "label", "memory", "vcpus", "pcidevs",
"maxmem", "kernel", "uses_default_kernel", "kernelopts", "uses_default_kernelopts", "maxmem", "kernel", "uses_default_kernel", "kernelopts", "uses_default_kernelopts",
"mac", "services", "include_in_backups", "debug", "drive" ) "mac", "services", "include_in_backups", "debug", "default_user", "qrexec_installed", "drive" )
for attribute in common_attr_list: for attribute in common_attr_list:
kwargs[attribute] = element.get(attribute) kwargs[attribute] = element.get(attribute)
@ -2857,6 +2879,9 @@ class QubesVmCollection(dict):
if "debug" in kwargs: if "debug" in kwargs:
kwargs["debug"] = True if kwargs["debug"] == "True" else False kwargs["debug"] = True if kwargs["debug"] == "True" else False
if "qrexec_installed" in kwargs:
kwargs["qrexec_installed"] = True if kwargs["qrexec_installed"] == "True" else False
if "drive" in kwargs and kwargs["drive"] == "None": if "drive" in kwargs and kwargs["drive"] == "None":
kwargs["drive"] = None kwargs["drive"] = None

View File

@ -78,6 +78,12 @@ def do_list(vm):
if hasattr(vm, 'debug'): if hasattr(vm, 'debug'):
print fmt.format("debug", "on" if vm.debug else "off") print fmt.format("debug", "on" if vm.debug else "off")
if hasattr(vm, 'default_user'):
print fmt.format("default user", str(vm.default_user))
if hasattr(vm, 'qrexec_installed'):
print fmt.format("qrexec_installed", str(vm.qrexec_installed))
if hasattr(vm, 'drive'): if hasattr(vm, 'drive'):
print fmt.format("drive", str(vm.drive)) print fmt.format("drive", str(vm.drive))
@ -311,6 +317,22 @@ def set_debug(vms, vm, args):
vm.debug = bool(eval(args[0].capitalize())) vm.debug = bool(eval(args[0].capitalize()))
return True return True
def set_default_user(vms, vm, args):
if len (args) != 1:
print >> sys.stderr, "Missing user name!"
return False
vm.default_user = args[0]
return True
def set_qrexec_installed(vms, vm, args):
if len (args) != 1:
print >> sys.stderr, "Missing value (True/False)!"
return False
vm.qrexec_installed = bool(eval(args[0].capitalize()))
return True
def set_timezone(vms, vm, args): def set_timezone(vms, vm, args):
if len (args) != 1: if len (args) != 1:
print >> sys.stderr, "Missing value ('localtime' or timeoffset in seconds)!" print >> sys.stderr, "Missing value ('localtime' or timeoffset in seconds)!"
@ -338,6 +360,8 @@ properties = {
"drive": set_drive, "drive": set_drive,
"mac": set_mac, "mac": set_mac,
"debug": set_debug, "debug": set_debug,
"default_user": set_default_user,
"qrexec_installed": set_qrexec_installed,
"timezone": set_timezone, "timezone": set_timezone,
} }

View File

@ -98,7 +98,7 @@ def main():
parser.add_option ("-q", "--quiet", action="store_false", dest="verbose", default=True) parser.add_option ("-q", "--quiet", action="store_false", dest="verbose", default=True)
parser.add_option ("-a", "--auto", action="store_true", dest="auto", default=False, parser.add_option ("-a", "--auto", action="store_true", dest="auto", default=False,
help="Auto start the VM if not running") help="Auto start the VM if not running")
parser.add_option ("-u", "--user", action="store", dest="user", default="user", parser.add_option ("-u", "--user", action="store", dest="user", default=None,
help="Run command in a VM as a specified user") help="Run command in a VM as a specified user")
parser.add_option ("--tray", action="store_true", dest="tray", default=False, parser.add_option ("--tray", action="store_true", dest="tray", default=False,
help="Use tray notifications instead of stdout" ) help="Use tray notifications instead of stdout" )
@ -188,12 +188,12 @@ def main():
exit(1) exit(1)
vms_list.append(vm) vms_list.append(vm)
for vm in vms_list:
if takes_cmd_argument: if takes_cmd_argument:
cmd = "{user}:{cmd}".format(user=options.user, cmd=cmdstr) cmd = "{user}:{cmd}".format(user=options.user if options.user else vm.default_user, cmd=cmdstr)
else: else:
cmd = None cmd = None
for vm in vms_list:
vm_run_cmd(vm, cmd, options) vm_run_cmd(vm, cmd, options)