Initial import from Sabayon (ver 0.9.9.56)

1 files changed, 326 insertions, 0 deletions

diff --git a/image.py b/image.py
new file mode 100644
index 0000000..60f03aa
--- /dev/null
+++ b/image.py
@@ -0,0 +1,326 @@
+#
+# image.py: Support methods for CD/DVD and ISO image installations.
+#
+# Copyright (C) 2007  Red Hat, Inc.  All rights reserved.
+#
+# 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, see <http://www.gnu.org/licenses/>.
+#
+
+import isys, iutil
+import os, os.path, stat, string, sys
+from constants import *
+
+import gettext
+_ = lambda x: gettext.ldgettext("anaconda", x)
+
+import logging
+log = logging.getLogger("anaconda")
+
+_arch = iutil.getArch()
+
+def findIsoImages(path, messageWindow):
+    flush = os.stat(path)
+    files = os.listdir(path)
+    arch = _arch
+    discImages = {}
+
+    for file in files:
+        what = path + '/' + file
+        if not isys.isIsoImage(what):
+            continue
+
+        try:
+            isys.losetup("/dev/loop2", what, readOnly = 1)
+        except SystemError:
+            continue
+
+        try:
+            isys.mount("/dev/loop2", "/mnt/cdimage", fstype = "iso9660",
+                       readOnly = True)
+            for num in range(1, 10):
+                if os.access("/mnt/cdimage/.discinfo", os.R_OK):
+                    f = open("/mnt/cdimage/.discinfo")
+                    try:
+                        f.readline() # skip timestamp
+                        f.readline() # skip release description
+                        discArch = string.strip(f.readline()) # read architecture
+                        discNum = getDiscNums(f.readline().strip())
+                    except:
+                        discArch = None
+                        discNum = [ 0 ]
+
+                    f.close()
+
+                    if num not in discNum or discArch != arch:
+                        continue
+
+                    # if it's disc1, it needs to have images/install.img
+                    if (num == 1 and not
+                        os.access("/mnt/cdimage/images/install.img", os.R_OK)):
+                        log.warning("%s doesn't have a install.img, skipping" %(what,))
+                        continue
+                    # we only install binary packages, so let's look for a
+                    # product/ dir and hope that this avoids getting
+                    # discs from the src.rpm set
+                    if not os.path.isdir("/mnt/cdimage/%s" %(productPath,)):
+                        log.warning("%s doesn't have binary RPMS, skipping" %(what,))
+                        continue
+
+                    # warn user if images appears to be wrong size
+                    if os.stat(what)[stat.ST_SIZE] % 2048:
+                        rc = messageWindow(_("Warning"),
+                             _("The ISO image %s has a size which is not "
+                               "a multiple of 2048 bytes.  This may mean "
+                               "it was corrupted on transfer to this computer."
+                               "\n\n"
+                               "It is recommended that you exit and abort your "
+                               "installation, but you can choose to continue if "
+                               "you think this is in error.") % (file,),
+                               type="custom", custom_icon="warning",
+                               custom_buttons= [_("_Exit installer"),
+                                                _("_Continue")])
+                        if rc == 0:
+                            sys.exit(0)
+
+                    discImages[num] = file
+
+            isys.umount("/mnt/cdimage", removeDir=False)
+        except SystemError:
+            pass
+
+        isys.unlosetup("/dev/loop2")
+
+    return discImages
+
+def getDiscNums(line):
+    # get the disc numbers for this disc
+    nums = line.split(",")
+    if nums == ['ALL']: # Treat "ALL" DVD as disc 1
+        return [1]
+    discNums = []
+    for num in nums:
+        discNums.append(int(num))
+    return discNums
+
+def getMediaId(path):
+    if os.access("%s/.discinfo" % path, os.R_OK):
+        f = open("%s/.discinfo" % path)
+        newStamp = f.readline().strip()
+        f.close()
+
+        return newStamp
+    else:
+        return None
+
+# This mounts the directory containing the iso images, and places the
+# mount point in /mnt/isodir.
+def mountDirectory(methodstr, messageWindow):
+    if methodstr.startswith("hd:"):
+        method = methodstr[3:]
+        if method.count(":") == 1:
+            (device, path) = method.split(":")
+            fstype = "auto"
+        else:
+            (device, fstype, path) = method.split(":")
+
+        if not device.startswith("/dev/") and not device.startswith("UUID=") \
+           and not device.startswith("LABEL="):
+            device = "/dev/%s" % device
+    elif methodstr.startswith("nfsiso:"):
+        device = methodstr[7:]
+        fstype = "nfs"
+    else:
+        return
+
+    # No need to mount it again.
+    if os.path.ismount("/mnt/isodir"):
+        return
+
+    while True:
+        try:
+            isys.mount(device, "/mnt/isodir", fstype = fstype)
+            break
+        except SystemError, msg:
+            log.error("couldn't mount ISO source directory: %s" % msg)
+            ans = messageWindow(_("Couldn't Mount ISO Source"),
+                          _("An error occurred mounting the source "
+                            "device %s.  This may happen if your ISO "
+                            "images are located on an advanced storage "
+                            "device like LVM or RAID, or if there was a "
+                            "problem mounting a partition.  Click exit "
+                            "to abort the installation.")
+                          % (device,), type="custom", custom_icon="error",
+                          custom_buttons=[_("_Exit"), _("_Retry")])
+
+            if ans == 0:
+                sys.exit(0)
+            else:
+                continue
+
+def mountImage(isodir, tree, discnum, messageWindow, discImages={}):
+    if os.path.ismount(tree):
+        raise SystemError, "trying to mount already-mounted iso image!"
+
+    if discImages == {}:
+        discImages = findIsoImages(isodir, messageWindow)
+
+    while True:
+        try:
+            isoImage = "%s/%s" % (isodir, discImages[discnum])
+            isys.losetup("/dev/loop1", isoImage, readOnly = 1)
+            isys.mount("/dev/loop1", tree, fstype = 'iso9660', readOnly = True)
+            break
+        except:
+            ans = messageWindow(_("Missing ISO 9660 Image"),
+                                _("The installer has tried to mount "
+                                  "image #%s, but cannot find it on "
+                                  "the hard drive.\n\n"
+                                  "Please copy this image to the "
+                                  "drive and click Retry.  Click Exit "
+                                  "to abort the installation.")
+                                  % (discnum,), type="custom",
+                                  custom_icon="warning",
+                                  custom_buttons=[_("_Exit"), _("_Retry")])
+            if ans == 0:
+                sys.exit(0)
+            elif ans == 1:
+                discImages = findIsoImages(isodir, messageWindow)
+
+    return discImages
+
+# given groupset containing information about selected packages, use
+# the disc number info in the headers to come up with message describing
+# the required CDs
+#
+# dialog returns a value of 0 if user selected to abort install
+def presentRequiredMediaMessage(anaconda):
+    reqcds = anaconda.backend.getRequiredMedia()
+
+    # if only one CD required no need to pop up a message
+    if len(reqcds) < 2:
+        return
+
+    # check what discs our currently mounted one provides
+    if os.access("%s/.discinfo" % anaconda.backend.ayum.tree, os.R_OK):
+        discNums = []
+        try:
+            f = open("%s/.discinfo" % anaconda.backend.ayum.tree)
+            stamp = f.readline().strip()
+            descr = f.readline().strip()
+            arch = f.readline().strip()
+            discNums = getDiscNums(f.readline().strip())
+            f.close()
+        except Exception, e:
+            log.critical("Exception reading discinfo: %s" %(e,))
+
+        log.info("discNums is %s" %(discNums,))
+        haveall = 0
+        s = set(reqcds)
+        t = set(discNums)
+        if s.issubset(t):
+            haveall = 1
+
+        if haveall == 1:
+            return
+
+    reqcds.sort()
+    reqcds = map(lambda disc: "#%s" % disc, filter(lambda disc: disc != -99, reqcds))
+    reqcdstr = ", ".join(reqcds)
+
+    return anaconda.intf.messageWindow(_("Required Install Media"),
+               _("The software you have selected to install will require the "
+                 "following %(productName)s %(productVersion)s "
+                 "discs:\n\n%(reqcdstr)s\nPlease have these ready "
+                 "before proceeding with the installation.  If you need to "
+                 "abort the installation and exit please select "
+                 "\"Reboot\".") % {'productName': product.productName,
+                                   'productVersion': product.productVersion,
+                                   'reqcdstr': reqcdstr},
+                 type="custom", custom_icon="warning",
+                 custom_buttons=[_("_Reboot"), _("_Back"), _("_Continue")])
+
+# Find an attached CD/DVD drive with media in it that contains packages,
+# and return that device name.
+def scanForMedia(tree, storage):
+    for dev in storage.devicetree.devices:
+        if dev.type != "cdrom":
+            continue
+
+        storage.devicetree.updateDeviceFormat(dev)
+        try:
+            dev.format.mount(mountpoint=tree)
+        except:
+            continue
+
+        if not verifyMedia(tree, 1):
+            dev.format.unmount()
+            continue
+
+        return dev.name
+
+    return None
+
+def umountImage(tree, currentMedia):
+    if currentMedia is not None:
+        isys.umount(tree, removeDir=False)
+        isys.unlosetup("/dev/loop1")
+
+def unmountCD(dev, messageWindow):
+    if not dev:
+        return
+
+    while True:
+        try:
+            dev.format.unmount()
+            break
+        except Exception, e:
+            log.error("exception in _unmountCD: %s" %(e,))
+            messageWindow(_("Error"),
+                          _("An error occurred unmounting the disc.  "
+                            "Please make sure you're not accessing "
+                            "%s from the shell on tty2 "
+                            "and then click OK to retry.")
+                          % (dev.path,))
+
+def verifyMedia(tree, discnum, timestamp=None):
+    if os.access("%s/.discinfo" % tree, os.R_OK):
+        f = open("%s/.discinfo" % tree)
+
+        newStamp = f.readline().strip()
+
+        try:
+            descr = f.readline().strip()
+        except:
+            descr = None
+
+        try:
+            arch = f.readline().strip()
+        except:
+            arch = None
+
+        try:
+            discs = getDiscNums(f.readline().strip())
+        except:
+            discs = [ 0 ]
+
+        f.close()
+
+        if timestamp is not None:
+            if newStamp == timestamp and arch == _arch and discnum in discs:
+                return True
+        else:
+            if arch == _arch and discnum in discs:
+                return True
+
+    return False