# __init__.py # Entry point for anaconda's software management module. # # Copyright (C) 2012 Red Hat, Inc. # # This copyrighted material is made available to anyone wishing to use, # modify, copy, or redistribute it subject to the terms and conditions of # the GNU General Public License v.2, or (at your option) any later version. # This program is distributed in the hope that it will be useful, but WITHOUT # ANY WARRANTY expressed or implied, including the implied warranties 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. Any Red Hat trademarks that are incorporated in the # source code or documentation are not subject to the GNU General Public # License and may only be used or replicated with the express permission of # Red Hat, Inc. # # Red Hat Author(s): David Lehman # Chris Lumens # """ TODO - error handling!!! - document all methods """ import os import requests import configparser import shutil from glob import glob from fnmatch import fnmatch import threading import re import functools from pyanaconda.iutil import requests_session from pyanaconda.iutil import open # pylint: disable=redefined-builtin if __name__ == "__main__": from pyanaconda import anaconda_log anaconda_log.init() from pyanaconda.constants import DRACUT_ISODIR, DRACUT_REPODIR, DD_ALL, DD_FIRMWARE, DD_RPMS, INSTALL_TREE, ISO_DIR from pyanaconda.constants import THREAD_STORAGE, THREAD_WAIT_FOR_CONNECTING_NM, THREAD_PAYLOAD from pyanaconda.constants import THREAD_PAYLOAD_RESTART from pykickstart.constants import GROUP_ALL, GROUP_DEFAULT, GROUP_REQUIRED from pyanaconda.flags import flags from pyanaconda.i18n import _, N_ from pyanaconda import iutil from pyanaconda import isys from pyanaconda.image import findFirstIsoImage from pyanaconda.image import mountImage from pyanaconda.image import opticalInstallMedia, verifyMedia from pyanaconda.iutil import ProxyString, ProxyStringError from pyanaconda.threads import threadMgr, AnacondaThread from pyanaconda.regexes import VERSION_DIGITS from pykickstart.parser import Group import logging log = logging.getLogger("packaging") from blivet.errors import StorageError import blivet.util import blivet.arch from blivet.platform import platform from blivet import setSysroot from pyanaconda.product import productName, productVersion USER_AGENT = "%s (anaconda)/%s" %(productName, productVersion) from distutils.version import LooseVersion REPO_NOT_SET = False def versionCmp(v1, v2): """ Compare two version number strings. """ firstVersion = LooseVersion(v1) secondVersion = LooseVersion(v2) return (firstVersion > secondVersion) - (firstVersion < secondVersion) ### ### ERROR HANDLING ### class PayloadError(Exception): pass class MetadataError(PayloadError): pass class NoNetworkError(PayloadError): pass # setup class PayloadSetupError(PayloadError): pass class ImageMissingError(PayloadSetupError): pass class ImageDirectoryMountError(PayloadSetupError): pass # software selection class NoSuchGroup(PayloadError): def __init__(self, group, adding=True, required=False): PayloadError.__init__(self, group) self.group = group self.adding = adding self.required = required class NoSuchPackage(PayloadError): def __init__(self, package, required=False): PayloadError.__init__(self, package) self.package = package self.required = required class DependencyError(PayloadError): pass # installation class PayloadInstallError(PayloadError): pass class Payload(object): """ Payload is an abstract class for OS install delivery methods. """ def __init__(self, data): """ data is a kickstart.AnacondaKSHandler class """ if self.__class__ is Payload: raise TypeError("Payload is an abstract class") self.data = data self.storage = None self.instclass = None self.txID = None # A list of verbose error strings from the subclass self.verbose_errors = [] self._session = requests_session() def setup(self, storage, instClass): """ Do any payload-specific setup. """ self.storage = storage self.instclass = instClass self.verbose_errors = [] def unsetup(self): """ Invalidate a previously setup paylaod. """ self.storage = None self.instclass = None def preStorage(self): """ Do any payload-specific work necessary before writing the storage configuration. This method need not be provided by all payloads. """ pass def release(self): """ Release any resources in use by this object, but do not do final cleanup. This is useful for dealing with payload backends that do not get along well with multithreaded programs. """ pass def reset(self): """ Reset the instance, not including ksdata. """ pass def prepareMountTargets(self, storage): """Run when physical storage is mounted, but other mount points may not exist. Used by the RPMOSTreePayload subclass. """ pass ### ### METHODS FOR WORKING WITH REPOSITORIES ### @property def addOns(self): """ A list of addon repo identifiers. """ return [r.name for r in self.data.repo.dataList()] @property def baseRepo(self): """ Get the identifier of the current base repo. or None """ return None @property def mirrorEnabled(self): """Is the closest/fastest mirror option enabled? This does not make sense for those payloads that do not support this concept. """ return True def isRepoEnabled(self, repo_id): """ Return True if repo is enabled. """ repo = self.getAddOnRepo(repo_id) if repo: return repo.enabled else: return False def getAddOnRepo(self, repo_id): """ Return a ksdata Repo instance matching the specified repo id. """ repo = None for r in self.data.repo.dataList(): if r.name == repo_id: repo = r break return repo def _repoNeedsNetwork(self, repo): """ Returns True if the ksdata repo requires networking. """ urls = [repo.baseurl] if repo.mirrorlist: urls.extend(repo.mirrorlist) network_protocols = ["http:", "ftp:", "nfs:", "nfsiso:"] for url in urls: if any(url.startswith(p) for p in network_protocols): return True return False @property def needsNetwork(self): return any(self._repoNeedsNetwork(r) for r in self.data.repo.dataList()) def updateBaseRepo(self, fallback=True, checkmount=True): """ Update the base repository from ksdata.method. """ pass def gatherRepoMetadata(self): pass def addRepo(self, newrepo): """Add the repo given by the pykickstart Repo object newrepo to the system. The repo will be automatically enabled and its metadata fetched. Duplicate repos will not raise an error. They should just silently take the place of the previous value. """ # Add the repo to the ksdata so it'll appear in the output ks file. self.data.repo.dataList().append(newrepo) def removeRepo(self, repo_id): repos = self.data.repo.dataList() try: idx = [repo.name for repo in repos].index(repo_id) except ValueError: log.error("failed to remove repo %s: not found", repo_id) else: repos.pop(idx) def enableRepo(self, repo_id): repo = self.getAddOnRepo(repo_id) if repo: repo.enabled = True def disableRepo(self, repo_id): repo = self.getAddOnRepo(repo_id) if repo: repo.enabled = False ### ### METHODS FOR WORKING WITH GROUPS ### def languageGroups(self): return [] def groupSelected(self, groupid): return Group(groupid) in self.data.packages.groupList def selectGroup(self, groupid, default=True, optional=False): if optional: include = GROUP_ALL elif default: include = GROUP_DEFAULT else: include = GROUP_REQUIRED grp = Group(groupid, include=include) if grp in self.data.packages.groupList: # I'm not sure this would ever happen, but ensure that re-selecting # a group with a different types set works as expected. if grp.include != include: grp.include = include return if grp in self.data.packages.excludedGroupList: self.data.packages.excludedGroupList.remove(grp) self.data.packages.groupList.append(grp) def deselectGroup(self, groupid): grp = Group(groupid) if grp in self.data.packages.excludedGroupList: return if grp in self.data.packages.groupList: self.data.packages.groupList.remove(grp) self.data.packages.excludedGroupList.append(grp) ### ### METHODS FOR QUERYING STATE ### @property def spaceRequired(self): """ The total disk space (Size) required for the current selection. """ raise NotImplementedError() @property def kernelVersionList(self): """ An iterable of the kernel versions installed by the payload. """ raise NotImplementedError() ## ## METHODS FOR TREE VERIFICATION ## def _getTreeInfo(self, url, proxy_url, sslverify): """ Retrieve treeinfo and return the path to the local file. :param baseurl: url of the repo :type baseurl: string :param proxy_url: Optional full proxy URL of or "" :type proxy_url: string :param sslverify: True if SSL certificate should be verified :type sslverify: bool :returns: Path to retrieved .treeinfo file or None :rtype: string or None """ if not url: return None log.debug("retrieving treeinfo from %s (proxy: %s ; sslverify: %s)", url, proxy_url, sslverify) proxies = {} if proxy_url: try: proxy = ProxyString(proxy_url) proxies = {"http": proxy.url, "https": proxy.url} except ProxyStringError as e: log.info("Failed to parse proxy for _getTreeInfo %s: %s", proxy_url, e) response = None headers = {"user-agent": USER_AGENT} try: response = self._session.get("%s/.treeinfo" % url, headers=headers, proxies=proxies, verify=sslverify) except requests.exceptions.RequestException as e: try: response = self._session.get("%s/treeinfo" % url, headers=headers, proxies=proxies, verify=sslverify) except requests.exceptions.RequestException as e: log.info("Error downloading treeinfo: %s", e) self.verbose_errors.append(str(e)) response = None if response: # write the local treeinfo file with open("/tmp/.treeinfo", "w") as f: f.write(response.text) # and also return the treeinfo contents as a string return response.text else: return None def _getReleaseVersion(self, url): """ Return the release version of the tree at the specified URL. """ try: version = re.match(VERSION_DIGITS, productVersion).group(1) except AttributeError: version = "rawhide" log.debug("getting release version from tree at %s (%s)", url, version) if hasattr(self.data.method, "proxy"): proxy = self.data.method.proxy else: proxy = None treeinfo = self._getTreeInfo(url, proxy, not flags.noverifyssl) if treeinfo: c = configparser.ConfigParser() c.read(treeinfo) try: # Trim off any -Alpha or -Beta version = re.match(VERSION_DIGITS, c.get("general", "version")).group(1) except AttributeError: version = "rawhide" except configparser.Error: pass log.debug("got a release version of %s", version) return version ## ## METHODS FOR MEDIA MANAGEMENT (XXX should these go in another module?) ## @staticmethod def _setupDevice(device, mountpoint): """ Prepare an install CD/DVD for use as a package source. """ log.info("setting up device %s and mounting on %s", device.name, mountpoint) # Is there a symlink involved? If so, let's get the actual path. # This is to catch /run/install/isodir vs. /mnt/install/isodir, for # instance. realMountpoint = os.path.realpath(mountpoint) if os.path.ismount(realMountpoint): mdev = blivet.util.get_mount_device(realMountpoint) if mdev: log.warning("%s is already mounted on %s", mdev, mountpoint) if mdev == device.path: return else: try: blivet.util.umount(realMountpoint) except OSError as e: log.error(str(e)) log.info("umount failed -- mounting on top of it") try: device.setup() device.format.setup(mountpoint=mountpoint) except StorageError as e: log.error("mount failed: %s", e) device.teardown(recursive=True) raise PayloadSetupError(str(e)) @staticmethod def _setupNFS(mountpoint, server, path, options): """ Prepare an NFS directory for use as a package source. """ log.info("mounting %s:%s:%s on %s", server, path, options, mountpoint) if os.path.ismount(mountpoint): dev = blivet.util.get_mount_device(mountpoint) _server, colon, _path = dev.partition(":") if colon == ":" and server == _server and path == _path: log.debug("%s:%s already mounted on %s", server, path, mountpoint) return else: log.debug("%s already has something mounted on it", mountpoint) try: blivet.util.umount(mountpoint) except OSError as e: log.error(str(e)) log.info("umount failed -- mounting on top of it") # mount the specified directory url = "%s:%s" % (server, path) if not options: options = "nolock" elif "nolock" not in options: options += ",nolock" try: blivet.util.mount(url, mountpoint, fstype="nfs", options=options) except OSError as e: raise PayloadSetupError(str(e)) ### ### METHODS FOR INSTALLING THE PAYLOAD ### def preInstall(self, packages=None, groups=None): """ Perform pre-installation tasks. """ iutil.mkdirChain(iutil.getSysroot() + "/root") self._writeModuleBlacklist() def install(self): """ Install the payload. """ raise NotImplementedError() def _writeModuleBlacklist(self): """ Copy modules from modprobe.blacklist= on cmdline to /etc/modprobe.d/anaconda-blacklist.conf so that modules will continue to be blacklisted when the system boots. """ if "modprobe.blacklist" not in flags.cmdline: return iutil.mkdirChain(iutil.getSysroot() + "/etc/modprobe.d") with open(iutil.getSysroot() + "/etc/modprobe.d/anaconda-blacklist.conf", "w") as f: f.write("# Module blacklists written by anaconda\n") for module in flags.cmdline["modprobe.blacklist"].split(): f.write("blacklist %s\n" % module) def _copyDriverDiskFiles(self): # Multiple driver disks may be loaded, so we need to glob for all # the firmware files in the common DD firmware directory for f in glob(DD_FIRMWARE+"/*"): try: shutil.copyfile(f, "%s/lib/firmware/" % iutil.getSysroot()) except IOError as e: log.error("Could not copy firmware file %s: %s", f, e.strerror) #copy RPMS for d in glob(DD_RPMS): shutil.copytree(d, iutil.getSysroot() + "/root/" + os.path.basename(d)) #copy modules and firmware into root's home directory if os.path.exists(DD_ALL): try: shutil.copytree(DD_ALL, iutil.getSysroot() + "/root/DD") except IOError as e: log.error("failed to copy driver disk files: %s", e.strerror) # XXX TODO: real error handling, as this is probably going to # prevent boot on some systems def recreateInitrds(self): """ Recreate the initrds by calling kernel-install This needs to be done after all configuration files have been written, since dracut depends on some of them. :returns: None """ for kernel in self.kernelVersionList: log.info("recreating initrd for %s", kernel) if not flags.imageInstall: iutil.execInSysroot("kernel-install", ["add", kernel, "/boot/vmlinuz-%s" % kernel]) else: # hostonly is not sensible for disk image installations # using /dev/disk/by-uuid/ is necessary due to disk image naming iutil.execInSysroot("dracut", ["-N", "--persistent-policy", "by-uuid", "-f", "/boot/initramfs-%s.img" % kernel, kernel]) def _setDefaultBootTarget(self): """ Set the default systemd target for the system. """ if not os.path.exists(iutil.getSysroot() + "/etc/systemd/system"): log.error("systemd is not installed -- can't set default target") return # If X was already requested we don't have to continue if self.data.xconfig.startX: return try: import rpm except ImportError: log.info("failed to import rpm -- not adjusting default runlevel") else: ts = rpm.TransactionSet(iutil.getSysroot()) # XXX one day this might need to account for anaconda's display mode if ts.dbMatch("provides", 'service(graphical-login)').count() and \ ts.dbMatch('provides', 'xorg-x11-server-Xorg').count() and \ not flags.usevnc: # We only manipulate the ksdata. The symlink is made later # during the config write out. self.data.xconfig.startX = True def dracutSetupArgs(self): args = [] try: import rpm except ImportError: pass else: iutil.resetRpmDb() ts = rpm.TransactionSet(iutil.getSysroot()) # Only add "rhgb quiet" on non-s390, non-serial installs if iutil.isConsoleOnVirtualTerminal() and \ (ts.dbMatch('provides', 'rhgb').count() or \ ts.dbMatch('provides', 'plymouth').count()): args.extend(["rhgb", "quiet"]) return args def postInstall(self): """ Perform post-installation tasks. """ # set default systemd target self._setDefaultBootTarget() # write out static config (storage, modprobe, keyboard, ??) # kickstart should handle this before we get here self._copyDriverDiskFiles() def writeStorageEarly(self): """Some packaging payloads require that the storage configuration be written out before doing installation. Right now, this is basically just the dnfpayload. Payloads should only implement one of these methods by overriding the unneeded one with a pass. """ if not flags.dirInstall: self.storage.write() def writeStorageLate(self): """Some packaging payloads require that the storage configuration be written out after doing installation. Right now, this is basically every payload except for dnf. Payloads should only implement one of these methods by overriding the unneeded one with a pass. """ if not flags.dirInstall: if iutil.getSysroot() != iutil.getTargetPhysicalRoot(): setSysroot(iutil.getTargetPhysicalRoot(), iutil.getSysroot()) # Now that we have the FS layout in the target, umount # things that were in the legacy sysroot, and put them in # the target root, except for the physical /. First, # unmount all target filesystems. self.storage.umountFilesystems() # Explicitly mount the root on the physical sysroot rootmnt = self.storage.mountpoints.get('/') rootmnt.setup() rootmnt.format.setup(options=rootmnt.format.options, chroot=iutil.getTargetPhysicalRoot()) self.prepareMountTargets(self.storage) # Everything else goes in the target root, including /boot # since the bootloader code will expect to find /boot # inside the chroot. self.storage.mountFilesystems(skipRoot=True) self.storage.write() # Inherit abstract methods from Payload # pylint: disable=abstract-method class ImagePayload(Payload): """ An ImagePayload installs an OS image to the target system. """ def __init__(self, data): if self.__class__ is ImagePayload: raise TypeError("ImagePayload is an abstract class") Payload.__init__(self, data) # Inherit abstract methods from ImagePayload # pylint: disable=abstract-method class ArchivePayload(ImagePayload): """ An ArchivePayload unpacks source archives onto the target system. """ def __init__(self, data): if self.__class__ is ArchivePayload: raise TypeError("ArchivePayload is an abstract class") ImagePayload.__init__(self, data) class PackagePayload(Payload): """ A PackagePayload installs a set of packages onto the target system. """ DEFAULT_REPOS = [productName.split('-')[0].lower(), "rawhide"] def __init__(self, data): if self.__class__ is PackagePayload: raise TypeError("PackagePayload is an abstract class") super(PackagePayload, self).__init__(data) self.install_device = None self._rpm_macros = [] self.requiredPackages = [] self.requiredGroups = [] # Used to determine which add-ons to display for each environment. # The dictionary keys are environment IDs. The dictionary values are two-tuples # consisting of lists of add-on group IDs. The first list is the add-ons specific # to the environment, and the second list is the other add-ons possible for the # environment. self._environmentAddons = {} def preInstall(self, packages=None, groups=None): super(PackagePayload, self).preInstall() # Set rpm-specific options # nofsync speeds things up at the risk of rpmdb data loss in a crash. # But if we crash mid-install you're boned anyway, so who cares? self.rpmMacros.append(('__dbi_htconfig', 'hash nofsync %{__dbi_other} %{__dbi_perms}')) if self.data.packages.excludeDocs: self.rpmMacros.append(('_excludedocs', '1')) if self.data.packages.instLangs is not None: # Use nil if instLangs is empty self.rpmMacros.append(('_install_langs', self.data.packages.instLangs or '%{nil}')) if flags.selinux: for d in ["/tmp/updates", "/etc/selinux/targeted/contexts/files", "/etc/security/selinux/src/policy", "/etc/security/selinux"]: f = d + "/file_contexts" if os.access(f, os.R_OK): self.rpmMacros.append(('__file_context_path', f)) break else: self.rpmMacros.append(('__file_context_path', '%{nil}')) # Add platform specific group groupid = iutil.get_platform_groupid() if groupid and groupid in self.groups: if isinstance(groups, list): log.info("Adding platform group %s", groupid) groups.append(groupid) else: log.warning("Could not add %s to groups, not a list.", groupid) elif groupid: log.warning("Platform group %s not available.", groupid) @property def kernelPackages(self): if "kernel" in self.data.packages.excludedList: return [] kernels = ["kernel"] if isys.isPaeAvailable(): kernels.insert(0, "kernel-PAE") # most ARM systems use platform-specific kernels if blivet.arch.isARM(): if platform.armMachine is not None: kernels = ["kernel-%s" % platform.armMachine] if isys.isLpaeAvailable(): kernels.insert(0, "kernel-lpae") return kernels @property def kernelVersionList(self): # Find all installed rpms that provide 'kernel' # If a PackagePayload is in use, rpm needs to be available try: import rpm except ImportError: raise PayloadError("failed to import rpm-python, cannot determine kernel versions") files = [] ts = rpm.TransactionSet(iutil.getSysroot()) mi = ts.dbMatch('providename', 'kernel') for hdr in mi: unicode_fnames = (f.decode("utf-8") for f in hdr.filenames) # Find all /boot/vmlinuz- files and strip off vmlinuz- files.extend((f.split("/")[-1][8:] for f in unicode_fnames if fnmatch(f, "/boot/vmlinuz-*") or fnmatch(f, "/boot/efi/EFI/%s/vmlinuz-*" % self.instclass.efi_dir))) return sorted(files, key=functools.cmp_to_key(versionCmp)) @property def rpmMacros(self): """A list of (name, value) pairs to define as macros in the rpm transaction.""" return self._rpm_macros @rpmMacros.setter def rpmMacros(self, value): self._rpm_macros = value def reset(self): self.reset_install_device() def reset_install_device(self): """ Unmount the previous base repo and reset the install_device """ # cdrom: install_device.teardown (INSTALL_TREE) # hd: umount INSTALL_TREE, install_device.teardown (ISO_DIR) # nfs: umount INSTALL_TREE # nfsiso: umount INSTALL_TREE, umount ISO_DIR if os.path.ismount(INSTALL_TREE) and not flags.testing: if self.install_device and \ blivet.util.get_mount_device(INSTALL_TREE) == self.install_device.path: self.install_device.teardown(recursive=True) else: blivet.util.umount(INSTALL_TREE) if os.path.ismount(ISO_DIR) and not flags.testing: if self.install_device and \ blivet.util.get_mount_device(ISO_DIR) == self.install_device.path: self.install_device.teardown(recursive=True) # The below code will fail when nfsiso is the stage2 source # But if we don't do this we may not be able to switch from # one nfsiso repo to another nfsiso repo. We need to have a # way to detect the stage2 state and work around it. # Commenting out the below is a hack for F18. FIXME #else: # # NFS # blivet.util.umount(ISO_DIR) self.install_device = None def _setupMedia(self, device): method = self.data.method if method.method == "harddrive": self._setupDevice(device, mountpoint=ISO_DIR) # check for ISO images in the newly mounted dir path = ISO_DIR if method.dir: path = os.path.normpath("%s/%s" % (path, method.dir)) # XXX it would be nice to streamline this when we're just setting # things back up after storage activation instead of having to # pretend we don't already know which ISO image we're going to # use image = findFirstIsoImage(path) if not image: device.teardown(recursive=True) raise PayloadSetupError("failed to find valid iso image") if path.endswith(".iso"): path = os.path.dirname(path) # this could already be set up the first time through if not os.path.ismount(INSTALL_TREE): # mount the ISO on a loop image = os.path.normpath("%s/%s" % (path, image)) mountImage(image, INSTALL_TREE) if not method.dir.endswith(".iso"): method.dir = os.path.normpath("%s/%s" % (method.dir, os.path.basename(image))) while method.dir.startswith("/"): # riduculous method.dir = method.dir[1:] # Check to see if the device is already mounted, in which case # we don't need to mount it again elif method.method == "cdrom" and \ blivet.util.get_mount_paths(device.path): return else: device.format.setup(mountpoint=INSTALL_TREE) def _setupInstallDevice(self, storage, checkmount): # XXX FIXME: does this need to handle whatever was set up by dracut? method = self.data.method sslverify = True url = None mirrorlist = None # See if we already have stuff mounted due to dracut isodev = blivet.util.get_mount_device(DRACUT_ISODIR) device = blivet.util.get_mount_device(DRACUT_REPODIR) if method.method == "harddrive": if method.biospart: log.warning("biospart support is not implemented") devspec = method.biospart else: devspec = method.partition needmount = True # See if we used this method for stage2, thus dracut left it if isodev and method.partition and method.partition in isodev \ and DRACUT_ISODIR in device: # Everything should be setup url = "file://" + DRACUT_REPODIR needmount = False # We don't setup an install_device here # because we can't tear it down isodevice = storage.devicetree.resolveDevice(devspec) if needmount: if not isodevice: raise PayloadSetupError("device for HDISO install %s does not exist" % devspec) self._setupMedia(isodevice) url = "file://" + INSTALL_TREE self.install_device = isodevice elif method.method == "nfs": # There are several possible scenarios here: # 1. dracut could have mounted both the nfs repo and an iso and used # the stage2 from inside the iso to boot from. # isodev and device will be set in this case. # 2. dracut could have mounted the nfs repo and used a stage2 from # the NFS mount w/o mounting the iso. # isodev will be None and device will be the nfs: path # 3. dracut did not mount the nfs (eg. stage2 came from elsewhere) # isodev and device are both None # 4. The repo may not contain an iso, in that case use it as is if isodev: path = iutil.parseNfsUrl('nfs:%s' % isodev)[2] # See if the dir holding the iso is what we want # and also if we have an iso mounted to /run/install/repo if path and path in isodev and DRACUT_ISODIR in device: # Everything should be setup url = "file://" + DRACUT_REPODIR else: # see if the nfs dir is mounted needmount = True if device: _options, host, path = iutil.parseNfsUrl('nfs:%s' % device) if method.server and method.server == host and \ method.dir and method.dir == path: needmount = False path = DRACUT_REPODIR if needmount: # Mount the NFS share on INSTALL_TREE. If it ends up # being nfsiso we will move the mountpoint to ISO_DIR. if method.dir.endswith(".iso"): nfsdir = os.path.dirname(method.dir) else: nfsdir = method.dir self._setupNFS(INSTALL_TREE, method.server, nfsdir, method.opts) path = INSTALL_TREE # check for ISO images in the newly mounted dir if method.dir.endswith(".iso"): # if the given URL includes a specific ISO image file, use it image_file = os.path.basename(method.dir) path = os.path.normpath("%s/%s" % (path, image_file)) image = findFirstIsoImage(path) # An image was found, mount it on INSTALL_TREE if image: if path.startswith(INSTALL_TREE): # move the INSTALL_TREE mount to ISO_DIR so we can # mount the contents of the iso there. # work around inability to move shared filesystems iutil.execWithRedirect("mount", ["--make-rprivate", "/"]) iutil.execWithRedirect("mount", ["--move", INSTALL_TREE, ISO_DIR]) # The iso is now under ISO_DIR path = ISO_DIR elif path.endswith(".iso"): path = os.path.dirname(path) # mount the ISO on a loop image = os.path.normpath("%s/%s" % (path, image)) mountImage(image, INSTALL_TREE) url = "file://" + INSTALL_TREE else: # Fall back to the mount path instead of a mounted iso url = "file://" + path elif method.method == "url": url = method.url mirrorlist = method.mirrorlist sslverify = not (method.noverifyssl or flags.noverifyssl) elif method.method == "cdrom" or (checkmount and not method.method): # Did dracut leave the DVD or NFS mounted for us? device = blivet.util.get_mount_device(DRACUT_REPODIR) # Check for valid optical media if we didn't boot from one if not verifyMedia(DRACUT_REPODIR): self.install_device = opticalInstallMedia(storage.devicetree) # Only look at the dracut mount if we don't already have a cdrom if device and not self.install_device: self.install_device = storage.devicetree.getDeviceByPath(device) url = "file://" + DRACUT_REPODIR if not method.method: # See if this is a nfs mount if ':' in device: # prepend nfs: to the url as that's what the parser # wants. Note we don't get options from this, but # that's OK for the UI at least. _options, host, path = iutil.parseNfsUrl("nfs:%s" % device) method.method = "nfs" method.server = host method.dir = path else: method.method = "cdrom" else: if self.install_device: if not method.method: method.method = "cdrom" self._setupMedia(self.install_device) url = "file://" + INSTALL_TREE elif method.method == "cdrom": raise PayloadSetupError("no usable optical media found") return url, mirrorlist, sslverify ### ### METHODS FOR WORKING WITH REPOSITORIES ### @property def repos(self): """A list of repo identifiers, not objects themselves.""" raise NotImplementedError() def addDriverRepos(self): """ Add driver repositories and packages """ # Drivers are loaded by anaconda-dracut, their repos are copied # into /run/install/DD-X where X is a number starting at 1. The list of # packages that were selected is in /run/install/dd_packages # Add repositories dir_num = 0 while True: dir_num += 1 repo = "/run/install/DD-%d/" % dir_num if not os.path.isdir(repo): break # Drivers may be under // or /DD-net/, but they can also be # in the top level of the DD repo. if os.path.isdir(repo+"DD-net"): repo += "DD-net" elif os.path.isdir(repo+blivet.arch.getArch()): repo += blivet.arch.getArch() # Run createrepo if there are rpms and no repodata if not os.path.isdir(repo+"/repodata"): rpms = glob(repo+"/*rpm") if not rpms: continue log.info("Running createrepo on %s", repo) iutil.execWithRedirect("createrepo_c", [repo]) ks_repo = self.data.RepoData(name="DD-%d" % dir_num, baseurl="file://"+repo, enabled=True) self.addRepo(ks_repo) # Add packages if not os.path.exists("/run/install/dd_packages"): return with open("/run/install/dd_packages", "r") as f: for line in f: package = line.strip() if package not in self.requiredPackages: self.requiredPackages.append(package) log.debug("required packages = %s", self.requiredPackages) @property def ISOImage(self): """ The location of a mounted ISO repo, or None. """ if not self.data.method.method == "harddrive": return None # This could either be mounted to INSTALL_TREE or on # DRACUT_ISODIR if dracut did the mount. dev = blivet.util.get_mount_device(INSTALL_TREE) if dev: return dev[len(ISO_DIR)+1:] dev = blivet.util.get_mount_device(DRACUT_ISODIR) if dev: return dev[len(DRACUT_ISODIR)+1:] return None ### ### METHODS FOR WORKING WITH ENVIRONMENTS ### @property def environments(self): raise NotImplementedError() def environmentHasOption(self, environmentid, grpid): raise NotImplementedError() def environmentOptionIsDefault(self, environmentid, grpid): raise NotImplementedError() def environmentDescription(self, environmentid): raise NotImplementedError() def selectEnvironment(self, environmentid, excluded=None): if environmentid not in self.environments: raise NoSuchGroup(environmentid) self.data.packages.environment = environmentid if excluded is None: excluded = [] def environmentGroups(self, environmentid, optional=True): raise NotImplementedError() @property def environmentAddons(self): return self._environmentAddons def _isGroupVisible(self, grp): raise NotImplementedError() def _groupHasInstallableMembers(self, grp): raise NotImplementedError() def _refreshEnvironmentAddons(self): log.info("Refreshing environmentAddons") self._environmentAddons = {} for environment in self.environments: self._environmentAddons[environment] = ([], []) # Determine which groups are specific to this environment and which other groups # are available in this environment. for grp in self.groups: if not self._groupHasInstallableMembers(grp): continue elif self.environmentHasOption(environment, grp): self._environmentAddons[environment][0].append(grp) elif self._isGroupVisible(grp): self._environmentAddons[environment][1].append(grp) ### ### METHODS FOR WORKING WITH GROUPS ### @property def groups(self): raise NotImplementedError() def groupDescription(self, groupid): raise NotImplementedError() class PayloadManager(object): """Framework for starting and watching the payload thread. This class defines several states, and events can be triggered upon reaching a state. Depending on whether a state has already been reached when a listener is added, the event code may be run in either the calling thread or the payload thread. The event code will block the payload thread regardless, so try not to run anything that takes a long time. All states except STATE_ERROR are expected to happen linearly, and adding a listener for a state that has already been reached or passed will immediately trigger that listener. For example, if the payload thread is currently in STATE_GROUP_MD, adding a listener for STATE_NETWORK will immediately run the code being added for STATE_NETWORK. The payload thread data should be accessed using the payloadMgr object, and the running thread can be accessed using threadMgr with the THREAD_PAYLOAD constant, if you need to wait for it or something. The thread should be started using payloadMgr.restartThread. """ STATE_START = 0 # Waiting on storage STATE_STORAGE = 1 # Waiting on network STATE_NETWORK = 2 # Downloading package metadata STATE_PACKAGE_MD = 3 # Downloading group metadata STATE_GROUP_MD = 4 # All done STATE_FINISHED = 5 # Error STATE_ERROR = -1 # Error strings ERROR_SETUP = N_("Failed to set up installation source") ERROR_MD = N_("Error downloading package metadata") ERROR_SOURCE = N_("No installation source available") def __init__(self): self._event_lock = threading.Lock() self._event_listeners = {} self._thread_state = self.STATE_START self._error = None # Initialize a list for each event state for event_id in range(self.STATE_ERROR, self.STATE_FINISHED + 1): self._event_listeners[event_id] = [] @property def error(self): return _(self._error) def addListener(self, event_id, func): """Add a listener for an event. :param int event_id: The event to listen for, one of the EVENT_* constants :param function func: An object to call when the event is reached """ # Check that the event_id is valid assert isinstance(event_id, int) assert event_id <= self.STATE_FINISHED assert event_id >= self.STATE_ERROR # Add the listener inside the lock in case we need to run immediately, # to make sure the listener isn't triggered twice with self._event_lock: self._event_listeners[event_id].append(func) # If an error event was requested, run it if currently in an error state if event_id == self.STATE_ERROR: if event_id == self._thread_state: func() # Otherwise, run if the requested event has already occurred elif event_id <= self._thread_state: func() def restartThread(self, storage, ksdata, payload, instClass, fallback=False, checkmount=True): """Start or restart the payload thread. This method starts a new thread to restart the payload thread, so this method's return is not blocked by waiting on the previous payload thread. If there is already a payload thread restart pending, this method has no effect. :param blivet.Blivet storage: The blivet storage instance :param kickstart.AnacondaKSHandler ksdata: The kickstart data instance :param packaging.Payload payload: The payload instance :param installclass.BaseInstallClass instClass: The install class instance :param bool fallback: Whether to fall back to the default repo in case of error :param bool checkmount: Whether to check for valid mounted media """ log.debug("Restarting payload thread") # If a restart thread is already running, don't start a new one if threadMgr.get(THREAD_PAYLOAD_RESTART): return # Launch a new thread so that this method can return immediately threadMgr.add(AnacondaThread(name=THREAD_PAYLOAD_RESTART, target=self._restartThread, args=(storage, ksdata, payload, instClass, fallback, checkmount))) def _restartThread(self, storage, ksdata, payload, instClass, fallback, checkmount): # Wait for the old thread to finish threadMgr.wait(THREAD_PAYLOAD) # Start a new payload thread threadMgr.add(AnacondaThread(name=THREAD_PAYLOAD, target=self._runThread, args=(storage, ksdata, payload, instClass, fallback, checkmount))) def _setState(self, event_id): # Update the current state log.debug("Updating payload thread state: %d", event_id) with self._event_lock: # Update the state within the lock to avoid a race with listeners # currently being added self._thread_state = event_id # Run any listeners for the new state for func in self._event_listeners[event_id]: func() def _runThread(self, storage, ksdata, payload, instClass, fallback, checkmount): # This is the thread entry # Set the initial state self._error = None self._setState(self.STATE_START) # Wait for storage self._setState(self.STATE_STORAGE) threadMgr.wait(THREAD_STORAGE) # Wait for network self._setState(self.STATE_NETWORK) # FIXME: condition for cases where we don't want network # (set and use payload.needsNetwork ?) threadMgr.wait(THREAD_WAIT_FOR_CONNECTING_NM) self._setState(self.STATE_PACKAGE_MD) payload.setup(storage, instClass) # If this is a non-package Payload, we're done if not isinstance(payload, PackagePayload): self._setState(self.STATE_FINISHED) return # Keep setting up package-based repositories # Download package metadata try: payload.updateBaseRepo(fallback=fallback, checkmount=checkmount) payload.addDriverRepos() except (OSError, PayloadError) as e: log.error("PayloadError: %s", e) self._error = self.ERROR_SETUP self._setState(self.STATE_ERROR) payload.unsetup() return # Gather the group data self._setState(self.STATE_GROUP_MD) payload.gatherRepoMetadata() payload.release() # Check if that failed if not payload.baseRepo: log.error("No base repo configured") self._error = self.ERROR_MD self._setState(self.STATE_ERROR) payload.unsetup() return self._setState(self.STATE_FINISHED) # Initialize the PayloadManager instance payloadMgr = PayloadManager()