Port kernel features support to fedora

data/anaconda.conf

@@ -299,6 +299,9 @@ password_policies =
  user (quality 1, length 6, empty)
  luks (quality 1, length 6)
 
+# Should kernel options be shown in the software selection spoke?
+show_kernel_options = True
+
 [License]
 # A path to EULA (if any)
 #

data/profile.d/rhel.conf

@@ -38,6 +38,7 @@ swap_is_recommended = True
 
 [User Interface]
 custom_stylesheet = /usr/share/anaconda/pixmaps/redhat.css
+show_kernel_options = True
 
 [License]
 eula = /usr/share/redhat-release/EULA

pyanaconda/core/configuration/ui.py

@@ -121,3 +121,7 @@ def _validate_policy_attributes(attrs):
 
  if "quality" not in attrs:
  raise ValueError("The minimal quality is not specified.")
+
+ @property
+ def show_kernel_options(self):
+ return self._get_option("show_kernel_options", bool)

pyanaconda/modules/payloads/payload/dnf/dnf.py

@@ -477,3 +477,11 @@ def post_install_with_tasks(self):
  dnf_manager=self.dnf_manager
  )
  ]
+
+ def match_available_packages(self, pattern):
+ """Find available packages that match the specified pattern.
+
+ :param pattern: a pattern for package names
+ :return: a list of matched package names
+ """
+ return self.dnf_manager.match_available_packages(pattern)

pyanaconda/modules/payloads/payload/dnf/dnf_interface.py

@@ -226,3 +226,11 @@ def GetRepoConfigurations(self) -> List[Structure]:
  return RepoConfigurationData.to_structure_list(
  self.implementation.get_repo_configurations()
  )
+
+ def MatchAvailablePackages(self, pattern: Str) -> List[Str]:
+ """Find available packages that match the specified pattern.
+
+ :param pattern: a pattern for package names
+ :return: a list of matched package names
+ """
+ return self.implementation.match_available_packages(pattern)

pyanaconda/ui/gui/spokes/software_selection.py

@@ -32,9 +32,13 @@
 from pyanaconda.ui.gui.spokes.lib.detailederror import DetailedErrorDialog
 from pyanaconda.ui.gui.spokes.lib.software_selection import GroupListBoxRow, SeparatorRow, \
  EnvironmentListBoxRow
+from pyanaconda.ui.gui.utils import escape_markup
 from pyanaconda.ui.lib.software import SoftwareSelectionCache, get_software_selection_status, \
  is_software_selection_complete, get_group_data, get_environment_data
 from pyanaconda.ui.lib.subscription import is_cdn_registration_required
+from pyanaconda.ui.lib.software import FEATURE_64K, KernelFeatures, get_kernel_from_properties, \
+ get_available_kernel_features, get_kernel_titles_and_descriptions
+from pyanaconda.core.configuration.anaconda import conf
 
 import gi
 gi.require_version("Gtk", "3.0")
@@ -94,6 +98,23 @@ def __init__(self, *args, **kwargs):
  Gtk.Scrollable.get_vadjustment(addon_viewport)
  )
 
+ # Display a group of options for selecting desired properties of a kernel
+ self._kernel_box = self.builder.get_object("kernelBox")
+ self._combo_kernel_page_size = self.builder.get_object("kernelPageSizeCombo")
+ self._label_kernel_page_size = self.builder.get_object("kernelPageSizeLabel")
+
+ # Normally I would create these in the .glade file but due to a bug they weren't
+ # created properly
+ self._model_kernel_page_size = Gtk.ListStore(str, str)
+
+ kernel_labels = get_kernel_titles_and_descriptions()
+ for i in ["4k", "64k"]:
+ self._model_kernel_page_size.append([i, "<b>%s</b>\n%s" %
+ (escape_markup(kernel_labels[i][0]),
+ escape_markup(kernel_labels[i][1]))])
+ self._combo_kernel_page_size.set_model(self._model_kernel_page_size)
+ self._available_kernels = get_available_kernel_features(self.payload.proxy)
+
  @property
  def _selection(self):
  """The packages selection."""
@@ -214,10 +235,12 @@ def refresh(self):
  # Create a new software selection cache.
  self._selection_cache = SoftwareSelectionCache(self.payload.proxy)
  self._selection_cache.apply_selection_data(self._selection)
+ self._available_kernels = get_available_kernel_features(self.payload.proxy)
 
  # Refresh up the UI.
  self._refresh_environments()
  self._refresh_groups()
+ self._refresh_kernel_features()
 
  # Set up the info bar.
  self.clear_info()
@@ -291,13 +314,50 @@ def _clear_listbox(self, listbox):
  listbox.remove(child)
  del child
 
+ def _refresh_kernel_features(self):
+ """Display options for selecting kernel features."""
+
+ # Only showing parts of kernel box relevant for current system.
+ self._available_kernels = get_available_kernel_features(self.payload.proxy)
+
+ show_kernels = False
+ if conf.ui.show_kernel_options:
+ for (_key, val) in self._available_kernels.items():
+ if val:
+ show_kernels = True
+ break
+
+ if show_kernels:
+ self._kernel_box.set_visible(True)
+ self._kernel_box.set_no_show_all(False)
+
+ # Arm 64k page size kernel combo
+ self._combo_kernel_page_size.set_visible(self._available_kernels[FEATURE_64K])
+ self._combo_kernel_page_size.set_no_show_all(not self._available_kernels[FEATURE_64K])
+ self._label_kernel_page_size.set_visible(self._available_kernels[FEATURE_64K])
+ self._label_kernel_page_size.set_no_show_all(not self._available_kernels[FEATURE_64K])
+ else:
+ # Hide the entire box.
+ self._kernel_box.set_visible(False)
+ self._kernel_box.set_no_show_all(True)
+
  def apply(self):
  """Apply the changes."""
  self._kickstarted = False
 
  selection = self._selection_cache.get_selection_data()
  log.debug("Setting new software selection: %s", selection)
 
+ # Select kernel
+ property_64k = self._available_kernels[FEATURE_64K] and \
+ self._combo_kernel_page_size.get_active_id() == FEATURE_64K
+ kernel_properties = KernelFeatures(property_64k)
+ kernel = get_kernel_from_properties(kernel_properties)
+ if kernel is not None and conf.ui.show_kernel_options:
+ log.debug("Selected kernel package: %s", kernel)
+ selection.packages.append(kernel)
+ selection.excluded_packages.append("kernel")
+
  self.payload.set_packages_selection(selection)
 
  hubQ.send_not_ready(self.__class__.__name__)

pyanaconda/ui/lib/software.py

@@ -15,13 +15,18 @@
 # License and may only be used or replicated with the express permission of
 # Red Hat, Inc.
 #
+from collections import namedtuple
+from blivet.arch import is_aarch64
+
 from pyanaconda.anaconda_loggers import get_module_logger
 from pyanaconda.core.i18n import _
 from pyanaconda.modules.common.structures.comps import CompsEnvironmentData, CompsGroupData
 from pyanaconda.modules.common.structures.packages import PackagesSelectionData
 
 log = get_module_logger(__name__)
 
+FEATURE_64K = "64k"
+KernelFeatures = namedtuple("KernelFeatures", ["page_size_64k"])
 
 def get_environment_data(dnf_proxy, environment_name):
  """Get the environment data.
@@ -96,6 +101,41 @@ def get_software_selection_status(dnf_proxy, selection, kickstarted=False):
  return environment_data.name
 
 
+def get_available_kernel_features(dnf_proxy):
+ """Returns a dictionary with that shows which kernels should be shown in the UI.
+ """
+ features = {
+ FEATURE_64K: is_aarch64() and any(dnf_proxy.MatchAvailablePackages("kernel-64k"))
+ }
+
+ return features
+
+
+def get_kernel_titles_and_descriptions():
+ """Returns a dictionary with descriptions and titles for different kernel options.
+ """
+ kernel_features = {
+ "4k": (_("4k"), _("More efficient memory usage in smaller environments")),
+ "64k": (_("64k"), _("System performance gains for memory-intensive workloads")),
+ }
+
+ return kernel_features
+
+
+def get_kernel_from_properties(features):
+ """Translates the selection of required properties into a kernel package name and returns it
+ or returns None if no properties were selected.
+ """
+ kernels = {
+ # ARM 64k Package Name
+ (False): None,
+ (True): "kernel-64k",
+ }
+
+ kernel_package = kernels[features[0]]
+ return kernel_package
+
+
 class SoftwareSelectionCache(object):
  """The cache of the user software selection.
 

pyanaconda/ui/tui/spokes/software_selection.py

@@ -23,9 +23,12 @@
  is_software_selection_complete, SoftwareSelectionCache, get_group_data, get_environment_data
 from pyanaconda.ui.tui.spokes import NormalTUISpoke
 from pyanaconda.core.threads import thread_manager
+from pyanaconda.ui.lib.software import FEATURE_64K, KernelFeatures, \
+ get_kernel_from_properties, get_available_kernel_features, get_kernel_titles_and_descriptions
 from pyanaconda.core.i18n import N_, _
 from pyanaconda.core.constants import THREAD_PAYLOAD, THREAD_CHECK_SOFTWARE, \
  THREAD_SOFTWARE_WATCHER, PAYLOAD_TYPE_DNF
+from pyanaconda.core.configuration.anaconda import conf
 
 from simpleline.render.containers import ListColumnContainer
 from simpleline.render.prompt import Prompt
@@ -69,6 +72,8 @@ def __init__(self, data, storage, payload):
 
  # Get the packages configuration.
  self._selection_cache = SoftwareSelectionCache(self.payload.proxy)
+ self._kernel_selection = None
+ self._available_kernels = None
 
  # Are we taking values (package list) from a kickstart file?
  self._kickstarted = flags.automatedInstall and self.payload.proxy.PackagesKickstarted
@@ -92,6 +97,9 @@ def _initialize(self):
  """Initialize the spoke in a separate thread."""
  thread_manager.wait(THREAD_PAYLOAD)
 
+ self._available_kernels = get_available_kernel_features(self.payload.proxy)
+ self._kernel_selection = dict.fromkeys(self._available_kernels, False)
+
  # Initialize and check the software selection.
  self._initialize_selection()
 
@@ -252,7 +260,8 @@ def input(self, args, key):
  self.data,
  self.storage,
  self.payload,
- self._selection_cache
+ self._selection_cache,
+ self._kernel_selection
  )
  ScreenHandler.push_screen_modal(spoke)
  self.apply()
@@ -269,6 +278,19 @@ def apply(self):
  selection = self._selection_cache.get_selection_data()
  log.debug("Setting new software selection: %s", selection)
 
+ # Processing chosen kernel
+ if conf.ui.show_kernel_options:
+ self._available_kernels = get_available_kernel_features(self.payload.proxy)
+ feature_64k = self._available_kernels[FEATURE_64K] and \
+ self._kernel_selection[FEATURE_64K]
+ features = KernelFeatures(feature_64k)
+ kernel = get_kernel_from_properties(features)
+ if kernel:
+ log.debug("Selected kernel package: %s", kernel)
+ selection.packages.append(kernel)
+ selection.excluded_packages.append("kernel")
+
+ log.debug("Setting new software selection: %s", self._selection)
  self.payload.set_packages_selection(selection)
 
  def execute(self):
@@ -297,11 +319,12 @@ class AdditionalSoftwareSpoke(NormalTUISpoke):
  """The spoke for choosing the additional software."""
  category = SoftwareCategory
 
- def __init__(self, data, storage, payload, selection_cache):
+ def __init__(self, data, storage, payload, selection_cache, kernel_selection):
  super().__init__(data, storage, payload)
  self.title = N_("Software selection")
  self._container = None
  self._selection_cache = selection_cache
+ self._kernel_selection = kernel_selection
 
  def refresh(self, args=None):
  """Refresh the screen."""
@@ -342,11 +365,81 @@ def _select_group(self, group):
  else:
  self._selection_cache.deselect_group(group)
 
+ def _show_kernel_features_screen(self, kernels):
+ """Returns True if at least one non-standard kernel is available.
+ """
+ if not conf.ui.show_kernel_options:
+ return False
+ for val in kernels.values():
+ if val:
+ return True
+ return False
+
  def input(self, args, key):
  if self._container.process_user_input(key):
  return InputState.PROCESSED_AND_REDRAW
- else:
- return super().input(args, key)
+ if key.lower() == Prompt.CONTINUE:
+ available_kernels = get_available_kernel_features(self.payload.proxy)
+ if self._show_kernel_features_screen(available_kernels):
+ spoke = KernelSelectionSpoke(self.data, self.storage, self.payload,
+ self._selection_cache, self._kernel_selection,
+ available_kernels)
+ ScreenHandler.push_screen_modal(spoke)
+ self.execute()
+ self.close()
+ return InputState.PROCESSED
+
+ return super().input(args, key)
+
+ def apply(self):
+ pass
+
+
+class KernelSelectionSpoke(NormalTUISpoke):
+ """A subspoke for selecting kernel features.
+ """
+ def __init__(self, data, storage, payload, selection_cache,
+ _kernel_selection, available_kernels):
+ super().__init__(data, storage, payload)
+ self.title = N_("Kernel Options")
+ self._container = None
+ self._selection_cache = selection_cache
+ self._kernel_selection = _kernel_selection
+ self._available_kernels = available_kernels
+
+ def refresh(self, args=None):
+ NormalTUISpoke.refresh(self)
+
+ # Retrieving translated UI strings
+ labels = get_kernel_titles_and_descriptions()
+
+ # Updating kernel availability
+ self._available_kernels = get_available_kernel_features(self.payload.proxy)
+ self._container = ListColumnContainer(2, columns_width=38, spacing=2)
+
+ # Rendering kernel checkboxes
+ for (name, val) in self._kernel_selection.items():
+ if not self._available_kernels[name]:
+ continue
+ (title, text) = labels[name]
+ widget = CheckboxWidget(title="%s" % title, text="%s" % text, completed=val)
+ self._container.add(widget, callback=self._set_kernel_callback, data=name)
+
+ self.window.add_with_separator(TextWidget(_("Kernel options")))
+ self.window.add_with_separator(self._container)
+
+ def _set_kernel_callback(self, data):
+ self._kernel_selection[data] = not self._kernel_selection[data]
+
+ def input(self, args, key):
+ if self._container.process_user_input(key):
+ return InputState.PROCESSED_AND_REDRAW
+
+ if key.lower() == Prompt.CONTINUE:
+ self.close()
+ return InputState.PROCESSED
+
+ return super().input(args, key)
 
  def apply(self):
  pass

tests/unit_tests/pyanaconda_tests/modules/payloads/payload/test_module_payload_dnf.py

@@ -1181,6 +1181,16 @@ def test_url_get_repo_configurations(self, publisher):
 
  assert self.interface.GetRepoConfigurations() == [expected]
 
+ def test_match_available_packages(self):
+ """Test the MatchAvailablePackages method."""
+ assert self.interface.MatchAvailablePackages("p") == []
+
+ dnf_manager = Mock(spec=DNFManager)
+ dnf_manager.match_available_packages.return_value = ["p1", "p2"]
+ self.module._dnf_manager = dnf_manager
+
+ assert self.interface.MatchAvailablePackages("p") == ["p1", "p2"]
+
 
 class DNFModuleTestCase(unittest.TestCase):
  """Test the DNF module."""