272ec190c7
I fixed a lot of bugs with copying files to the initrd, and getting the symlinks. Also I moved some of the code around, to get a clearer flow.
150 lines
4.9 KiB
Python
150 lines
4.9 KiB
Python
#
|
|
# instroot.py
|
|
# install root class
|
|
#
|
|
# Copyright (C) 2009 Red Hat, Inc.
|
|
#
|
|
# 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/>.
|
|
#
|
|
# Red Hat Author(s): Martin Gracik <mgracik@redhat.com>
|
|
#
|
|
|
|
import os
|
|
import glob
|
|
|
|
from utils.fileutils import copy
|
|
|
|
|
|
class InstallTree(object):
|
|
|
|
def __init__(self, config, yum, output):
|
|
self.conf = config
|
|
self.yum = yum
|
|
|
|
self.so, self.se = output
|
|
|
|
def get_packages(self):
|
|
required = ["anaconda", "anaconda-runtime", "/etc/gtk-2.0/gtkrc"]
|
|
|
|
# kernel packages
|
|
required.extend(["kernel", "*firmware*"])
|
|
|
|
# add the XEN kernel package
|
|
if self.conf.buildarch == "i386":
|
|
required.append("kernel-PAE")
|
|
|
|
# get additional packages from the configuration files
|
|
packages_files = []
|
|
packages_files.append(os.path.join(self.conf.confdir, "packages",
|
|
"packages.all")),
|
|
packages_files.append(os.path.join(self.conf.confdir, "packages",
|
|
"packages.%s" % (self.conf.buildarch,)))
|
|
|
|
packages = set()
|
|
for file in packages_files:
|
|
if os.path.isfile(file):
|
|
try:
|
|
f = open(file, "r")
|
|
except IOError as why:
|
|
self.se.error("Unable to read packages configuration:"
|
|
" %s" % (why,))
|
|
else:
|
|
for line in f.readlines():
|
|
line, sep, comment = line.partition("#")
|
|
line = line.strip()
|
|
|
|
if not line:
|
|
continue
|
|
|
|
if line.startswith("-"):
|
|
packages.discard(line[1:])
|
|
else:
|
|
packages.add(line)
|
|
|
|
f.close()
|
|
|
|
required.extend(list(packages))
|
|
|
|
# logos
|
|
required.extend(["%s-logos" % (self.conf.product.lower(),),
|
|
"%s-release" % (self.conf.product.lower(),)])
|
|
|
|
return required
|
|
|
|
def add_packages(self, packages):
|
|
for package in packages:
|
|
ok = self.yum.add_package(package)
|
|
if not ok:
|
|
self.se.warning("No package '%s' available" (package,))
|
|
|
|
def install_packages(self):
|
|
# XXX why do we need this?
|
|
os.symlink(os.path.join("/", "tmp"),
|
|
os.path.join(self.conf.treedir, "var", "lib", "xkb"))
|
|
|
|
self.yum.install()
|
|
|
|
def copy_updates(self):
|
|
if self.conf.updates and os.path.isdir(self.conf.updates):
|
|
copy(src_root=self.conf.updates, src_path="*",
|
|
dst_root=self.conf.treedir, dst_path="")
|
|
|
|
self.conf.delAttr("updates")
|
|
|
|
def fix_udev_links(self):
|
|
# these links are broken by default (at least on i386)
|
|
for filename in ("udevcontrol", "udevsettle", "udevtrigger"):
|
|
filename = os.path.join(self.conf.treedir, "sbin", filename)
|
|
if os.path.islink(filename):
|
|
os.unlink(filename)
|
|
os.symlink("udevadm", filename)
|
|
|
|
def remove_modules_broken_links(self):
|
|
# remove build and source links from modules directories
|
|
build = os.path.join(self.conf.treedir, "lib", "modules", "*", "build")
|
|
build_files = glob.glob(build)
|
|
|
|
source = os.path.join(self.conf.treedir, "lib", "modules", "*", "source")
|
|
source_files = glob.glob(source)
|
|
|
|
[os.unlink(filename) for filename in build_files + source_files
|
|
if os.path.islink(filename)]
|
|
|
|
def get_kernelfiles(self):
|
|
kerneldir = os.path.join(self.conf.treedir, "boot")
|
|
|
|
if self.conf.buildarch == "ia64":
|
|
kerneldir = os.path.join(kerneldir, "efi", "EFI", "redhat")
|
|
|
|
return glob.glob(os.path.join(kerneldir, "vmlinuz-*"))
|
|
|
|
def run(self):
|
|
self.so.info("Getting the list of packages")
|
|
packages = self.get_packages()
|
|
|
|
self.so.info("Running yum")
|
|
self.add_packages(packages)
|
|
self.install_packages()
|
|
|
|
self.so.info("Copying the updates")
|
|
self.copy_updates()
|
|
|
|
self.so.info("Fixing udev links")
|
|
self.fix_udev_links()
|
|
|
|
self.so.info("Removing build and source links in modules directories")
|
|
self.remove_modules_broken_links()
|
|
|
|
return self.get_kernelfiles()
|