34befb9cb8
Instead of parsing hwinfo output, which might or might be present on systems, just parse the /proc/cpuinfo file, it's much more reliable.
111 lines
3.6 KiB
Python
111 lines
3.6 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
|
|
import shutil
|
|
from collections import OrderedDict
|
|
from libcalamares.utils import check_chroot_call
|
|
|
|
def cpuinfo():
|
|
''' Return the information in /proc/cpuinfo
|
|
as a dictionary in the following format:
|
|
cpu_info['proc0']={...}
|
|
cpu_info['proc1']={...}
|
|
|
|
'''
|
|
|
|
cpuinfo=OrderedDict()
|
|
procinfo=OrderedDict()
|
|
|
|
nprocs = 0
|
|
with open('/proc/cpuinfo') as f:
|
|
for line in f:
|
|
if not line.strip():
|
|
# end of one processor
|
|
cpuinfo['proc%s' % nprocs] = procinfo
|
|
nprocs=nprocs+1
|
|
# Reset
|
|
procinfo=OrderedDict()
|
|
else:
|
|
if len(line.split(':')) == 2:
|
|
procinfo[line.split(':')[0].strip()] = line.split(':')[1].strip()
|
|
else:
|
|
procinfo[line.split(':')[0].strip()] = ''
|
|
|
|
return cpuinfo
|
|
|
|
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 = cpu_info()
|
|
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['proc0']['vendor_id'].lower() != "genuineintel":
|
|
modules.append("crc32c")
|
|
elif btrfs is "yes" and cpu['proc0']['vendor_id'].lower() == "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
|