#!/usr/bin/python2 # # The Qubes OS Project, http://www.qubes-os.org # # Copyright (C) 2010 Joanna Rutkowska # Copyright (C) 2013 Marek Marczykowski # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 2 # of the License, or (at your option) any later version. # # 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 General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. # # import subprocess import sys import os import os.path import shutil import dbus from qubes.qubes import QubesVm, QubesHVm from qubes.qubes import QubesException, QubesHost, QubesVmLabels from qubes.qubes import vm_files, system_path, dry_run import qubes.imgconverter vm_files['appmenus_templates_subdir'] = 'apps.templates' vm_files['appmenus_template_icons_subdir'] = 'apps.tempicons' vm_files['appmenus_icons_subdir'] = 'apps.icons' vm_files['appmenus_template_templates_subdir'] = 'apps-template.templates' vm_files['appmenus_whitelist'] = 'whitelisted-appmenus.list' system_path['appmenu_start_hvm_template'] = \ '/usr/share/qubes-appmenus/qubes-start.desktop' system_path['appmenu_create_cmd'] = \ '/usr/libexec/qubes-appmenus/create-apps-for-appvm.sh' system_path['appmenu_remove_cmd'] = \ '/usr/libexec/qubes-appmenus/remove-appvm-appmenus.sh' def QubesVm_get_attrs_config(self, attrs): attrs["appmenus_templates_dir"] = { "func": lambda x: os.path.join(self.dir_path, vm_files["appmenus_templates_subdir"]) if self.updateable else (self.template.appmenus_templates_dir if self.template is not None else None) } attrs["appmenus_template_icons_dir"] = { "func": lambda x: os.path.join(self.dir_path, vm_files["appmenus_template_icons_subdir"]) if self.updateable else (self.template.appmenus_template_icons_dir if self.template is not None else None) } attrs["appmenus_icons_dir"] = { "func": lambda x: os.path.join(self.dir_path, vm_files["appmenus_icons_subdir"])} return attrs def QubesTemplateVm_get_attrs_config(self, attrs): attrs['appmenus_templates_dir'] = { 'func': lambda x: os.path.join(self.dir_path, vm_files["appmenus_templates_subdir"])} attrs['appmenus_template_icons_dir'] = { 'func': lambda x: os.path.join(self.dir_path, vm_files["appmenus_template_icons_subdir"])} return attrs def QubesVm_appmenus_create(self, verbose=False, source_template=None): if source_template is None: source_template = self.template if self.internal: return if self.is_netvm(): vmtype = 'servicevms' elif self.is_template(): vmtype = 'vm-templates' else: vmtype = 'appvms' try: msgoutput = None if verbose else open(os.devnull, 'w') if source_template is not None: subprocess.check_call([system_path["appmenu_create_cmd"], source_template.appmenus_templates_dir, self.name, vmtype, self.label.icon], stdout=msgoutput, stderr=msgoutput) elif self.appmenus_templates_dir is not None: subprocess.check_call([system_path["appmenu_create_cmd"], self.appmenus_templates_dir, self.name, vmtype, self.label.icon], stdout=msgoutput, stderr=msgoutput) else: # Only add apps to menu subprocess.check_call([system_path["appmenu_create_cmd"], "none", self.name, vmtype, self.label.icon], stdout=msgoutput, stderr=msgoutput) except subprocess.CalledProcessError: print >> sys.stderr, "Ooops, there was a problem creating appmenus " \ "for {0} VM!".format(self.name) def QubesVm_appmenus_remove(self): if self.is_netvm(): vmtype = 'servicevms' elif self.is_template(): vmtype = 'vm-templates' else: vmtype = 'appvms' subprocess.check_call([system_path["appmenu_remove_cmd"], self.name, vmtype], stderr=open(os.devnull, 'w')) def QubesVm_appicons_create(self, srcdir=None): if srcdir is None: srcdir = self.appmenus_template_icons_dir if srcdir is None: return if not os.path.exists(srcdir): return whitelist = os.path.join(self.dir_path, vm_files['appmenus_whitelist']) if os.path.exists(whitelist): whitelist = [line.strip() for line in open(whitelist)] else: whitelist = None if not os.path.exists(self.appmenus_icons_dir): os.mkdir(self.appmenus_icons_dir) elif not os.path.isdir(self.appmenus_icons_dir): os.unlink(self.appmenus_icons_dir) os.mkdir(self.appmenus_icons_dir) for icon in os.listdir(srcdir): desktop = os.path.splitext(icon)[0] + '.desktop' if whitelist and desktop not in whitelist: continue src_icon = os.path.join(srcdir, icon) dst_icon = os.path.join(self.appmenus_icons_dir, icon) if not os.path.exists(dst_icon) or \ os.path.getmtime(src_icon) > os.path.getmtime(dst_icon): qubes.imgconverter.tint(src_icon, dst_icon, self.label.color) def QubesVm_appicons_remove(self): if not os.path.exists(self.appmenus_icons_dir): return for icon in os.listdir(self.appmenus_icons_dir): os.unlink(os.path.join(self.appmenus_icons_dir, icon)) def QubesVm_appicons_cleanup(self): srcdir = self.appmenus_template_icons_dir if srcdir is None: return if not os.path.exists(srcdir): return for icon in os.listdir(self.appmenus_icons_dir): if not os.path.exists(os.path.join(srcdir, icon)): os.unlink(os.path.join(self.appmenus_icons_dir, icon)) def QubesVm_pre_rename(self, new_name): self.appmenus_remove() def QubesVm_post_rename(self, old_name): old_dirpath = os.path.join(os.path.dirname(self.dir_path), old_name) if self.appmenus_templates_dir is not None: self.appmenus_templates_dir = self.appmenus_templates_dir.replace( old_dirpath, self.dir_path) self.appmenus_create() def QubesVm_create_on_disk(self, verbose, source_template): if isinstance(self, QubesHVm) and source_template is None: if verbose: print >> sys.stderr, "--> Creating appmenus directory: {0}".format( self.appmenus_templates_dir) os.mkdir(self.appmenus_templates_dir) shutil.copy(system_path["appmenu_start_hvm_template"], self.appmenus_templates_dir) source_whitelist_filename = 'vm-' + vm_files["appmenus_whitelist"] if self.is_netvm(): source_whitelist_filename = 'netvm-' + vm_files["appmenus_whitelist"] if source_template and os.path.exists( os.path.join(source_template.dir_path, source_whitelist_filename)): if verbose: print >> sys.stderr, "--> Creating default whitelisted apps list: {0}". \ format(self.dir_path + '/' + vm_files["whitelisted_appmenus"]) shutil.copy( os.path.join(source_template.dir_path, source_whitelist_filename), os.path.join(self.dir_path, vm_files["whitelisted_appmenus"])) if source_template and self.updateable: if verbose: print >> sys.stderr, "--> Copying the template's appmenus templates dir:\n{0} ==>\n{1}". \ format(source_template.appmenus_templates_dir, self.appmenus_templates_dir) if os.path.isdir(source_template.appmenus_templates_dir): shutil.copytree(source_template.appmenus_templates_dir, self.appmenus_templates_dir) else: os.mkdir(self.appmenus_templates_dir) if os.path.isdir(source_template.appmenus_template_icons_dir): shutil.copytree(source_template.appmenus_template_icons_dir, self.appmenus_template_icons_dir) else: os.mkdir(self.appmenus_template_icons_dir) # Create appmenus self.appicons_create() self.appmenus_create(verbose=verbose) def QubesVm_clone_disk_files(self, src_vm, verbose): if src_vm.updateable and src_vm.appmenus_templates_dir is not None and \ self.appmenus_templates_dir is not None: if verbose: print >> sys.stderr, "--> Copying the template's appmenus templates dir:\n{0} ==>\n{1}". \ format(src_vm.appmenus_templates_dir, self.appmenus_templates_dir) shutil.copytree(src_vm.appmenus_templates_dir, self.appmenus_templates_dir) if src_vm.updateable and src_vm.appmenus_template_icons_dir is not None \ and self.appmenus_template_icons_dir is not None and \ os.path.isdir(src_vm.appmenus_template_icons_dir): if verbose: print >> sys.stderr, "--> Copying the template's appmenus " \ "template icons dir:\n{0} ==>\n{1}". \ format(src_vm.appmenus_template_icons_dir, self.appmenus_template_icons_dir) shutil.copytree(src_vm.appmenus_template_icons_dir, self.appmenus_template_icons_dir) for whitelist in ( vm_files["appmenus_whitelist"], 'vm-' + vm_files["appmenus_whitelist"], 'netvm-' + vm_files["appmenus_whitelist"]): if os.path.exists(os.path.join(src_vm.dir_path, whitelist)): if verbose: print >> sys.stderr, "--> Copying whitelisted apps list: {0}". \ format(whitelist) shutil.copy(os.path.join(src_vm.dir_path, whitelist), os.path.join(self.dir_path, whitelist)) # Create appmenus self.appicons_create() self.appmenus_create(verbose=verbose) def QubesVm_remove_from_disk(self): self.appmenus_remove() def QubesVm_label_setter(self, _): self.appicons_create() # Apparently desktop environments heavily caches the icons, # see #751 for details if os.environ.get("DESKTOP_SESSION", "") == "kde-plasma": try: os.unlink(os.path.expandvars( "$HOME/.kde/cache-$HOSTNAME/icon-cache.kcache")) except: pass try: notify_object = dbus.SessionBus().get_object( "org.freedesktop.Notifications", "/org/freedesktop/Notifications") notify_object.Notify( "Qubes", 0, self.label.icon, "Qubes", "You will need to log off and log in again for the VM icons " "to update in the KDE launcher menu", [], [], 10000, dbus_interface="org.freedesktop.Notifications") except: pass elif os.environ.get("DESKTOP_SESSION", "") == "xfce": self.appmenus_remove() self.appmenus_create() def QubesVm_appmenus_recreate(self): """ Force recreation of all appmenus and icons. For example when VM label color was changed """ self.appmenus_remove() self.appicons_remove() self.appicons_create() self.appmenus_create() def QubesVm_appmenus_update(self): """ Similar to appmenus_recreate, but do not touch unchanged files """ self.appmenus_remove() self.appicons_create() self.appicons_cleanup() self.appmenus_create() def QubesVm_set_attr(self, name, newvalue, oldvalue): if name == 'internal': if newvalue and not oldvalue: self.appmenus_remove() elif not newvalue and oldvalue: self.appmenus_create() # new methods QubesVm.appmenus_create = QubesVm_appmenus_create QubesVm.appmenus_remove = QubesVm_appmenus_remove QubesVm.appmenus_recreate = QubesVm_appmenus_recreate QubesVm.appmenus_update = QubesVm_appmenus_update QubesVm.appicons_create = QubesVm_appicons_create QubesVm.appicons_cleanup = QubesVm_appicons_cleanup QubesVm.appicons_remove = QubesVm_appicons_remove # hooks for existing methods QubesVm.hooks_get_attrs_config.append(QubesVm_get_attrs_config) QubesVm.hooks_pre_rename.append(QubesVm_pre_rename) QubesVm.hooks_post_rename.append(QubesVm_post_rename) QubesVm.hooks_create_on_disk.append(QubesVm_create_on_disk) QubesVm.hooks_clone_disk_files.append(QubesVm_clone_disk_files) QubesVm.hooks_remove_from_disk.append(QubesVm_remove_from_disk) QubesVm.hooks_label_setter.append(QubesVm_label_setter) QubesVm.hooks_set_attr.append(QubesVm_set_attr)