93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
# encoding: utf-8
|
|
# === This file is part of Calamares - <http://github.com/calamares> ===
|
|
#
|
|
# Copyright 2014, Philip Müller <philm@manjaro.org>
|
|
#
|
|
# Calamares 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 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Calamares 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 Calamares. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import libcalamares
|
|
import os
|
|
import subprocess
|
|
from libcalamares.utils import check_chroot_call
|
|
|
|
|
|
def get_cpu():
|
|
""" Check if system is an intel system. """
|
|
|
|
process1 = subprocess.Popen(["hwinfo", "--cpu"], stdout=subprocess.PIPE)
|
|
process2 = subprocess.Popen(["grep", "Model:[[:space:]]"],
|
|
stdin=process1.stdout, stdout=subprocess.PIPE)
|
|
process1.stdout.close()
|
|
out, err = process2.communicate()
|
|
return out.decode().lower()
|
|
|
|
|
|
def set_mkinitcpio_hooks_and_modules(hooks, modules, root_mount_point):
|
|
""" Set up mkinitcpio.conf """
|
|
|
|
with open("/etc/mkinitcpio.conf", "r") as mkinitcpio_file:
|
|
mklins = [x.strip() for x in mkinitcpio_file.readlines()]
|
|
|
|
for i in range(len(mklins)):
|
|
if mklins[i].startswith("HOOKS"):
|
|
mklins[i] = 'HOOKS="%s"' % ' '.join(hooks)
|
|
elif mklins[i].startswith("MODULES"):
|
|
mklins[i] = 'MODULES="%s"' % ' '.join(modules)
|
|
|
|
path = os.path.join(root_mount_point, "etc/mkinitcpio.conf")
|
|
with open(path, "w") as mkinitcpio_file:
|
|
mkinitcpio_file.write("\n".join(mklins) + "\n")
|
|
|
|
def modify_mkinitcpio_conf(partitions, root_mount_point):
|
|
""" Modifies mkinitcpio.conf """
|
|
|
|
cpu = get_cpu()
|
|
swap_uuid = ""
|
|
btrfs = ""
|
|
hooks = ["base", "udev", "autodetect", "modconf", "block", "keyboard", "keymap"]
|
|
modules = []
|
|
|
|
# It is important that the plymouth hook comes before any encrypt hook
|
|
plymouth_bin = os.path.join(root_mount_point, "usr/bin/plymouth")
|
|
if os.path.exists(plymouth_bin):
|
|
hooks.append("plymouth")
|
|
|
|
for partition in partitions:
|
|
if partition["fs"] == "linuxswap":
|
|
swap_uuid = partition["uuid"]
|
|
if partition["fs"] == "btrfs":
|
|
btrfs = "yes"
|
|
|
|
if swap_uuid is not "":
|
|
hooks.extend(["resume", "filesystems"])
|
|
else:
|
|
hooks.extend(["filesystems"])
|
|
|
|
if btrfs is "yes" and cpu is not "genuineintel":
|
|
modules.append("crc32c")
|
|
elif btrfs is "yes" and cpu is "genuineintel":
|
|
modules.append("crc32c-intel")
|
|
else:
|
|
hooks.append("fsck")
|
|
|
|
set_mkinitcpio_hooks_and_modules(hooks, modules, root_mount_point)
|
|
|
|
|
|
def run():
|
|
partitions = libcalamares.globalstorage.value("partitions")
|
|
root_mount_point = libcalamares.globalstorage.value("rootMountPoint")
|
|
modify_mkinitcpio_conf(partitions, root_mount_point)
|
|
return None
|