REDROOM

Navigation

Breadcrumbs

Create

Preview: plugin_irqbalance.py Size: 3.47 KB

/lib/python3.6/site-packages/tuned/plugins/plugin_irqbalance.py

from . import base
from .decorators import command_custom
from tuned import consts
import tuned.logs
import errno
import perf
import re

log = tuned.logs.get()

class IrqbalancePlugin(base.Plugin):
	"""
	`irqbalance`::
	
	Plug-in for irqbalance settings management. The plug-in
	configures CPUs which should be skipped when rebalancing IRQs in
	`/etc/sysconfig/irqbalance`. It then restarts irqbalance if and
	only if it was previously running.
	+
	The banned/skipped CPUs are specified as a CPU list via the
	[option]`banned_cpus` option.
	+
	.Skip CPUs 2,4 and 9-13 when rebalancing IRQs
	====
	----
	[irqbalance]
	banned_cpus=2,4,9-13
	----
	====
	"""

	def __init__(self, *args, **kwargs):
		super(IrqbalancePlugin, self).__init__(*args, **kwargs)
		self._cpus = perf.cpu_map()

	def _instance_init(self, instance):
		instance._has_dynamic_tuning = False
		instance._has_static_tuning = True

	def _instance_cleanup(self, instance):
		pass

	@classmethod
	def _get_config_options(cls):
		return {
			"banned_cpus": None,
		}

	def _read_irqbalance_sysconfig(self):
		try:
			with open(consts.IRQBALANCE_SYSCONFIG_FILE, "r") as f:
				return f.read()
		except IOError as e:
			if e.errno == errno.ENOENT:
				log.warn("irqbalance sysconfig file is missing. Is irqbalance installed?")
			else:
				log.error("Failed to read irqbalance sysconfig file: %s" % e)
			return None

	def _write_irqbalance_sysconfig(self, content):
		try:
			with open(consts.IRQBALANCE_SYSCONFIG_FILE, "w") as f:
				f.write(content)
			return True
		except IOError as e:
			log.error("Failed to write irqbalance sysconfig file: %s" % e)
			return False

	def _write_banned_cpus(self, sysconfig, banned_cpumask):
		return sysconfig + "IRQBALANCE_BANNED_CPUS=%s\n" % banned_cpumask

	def _clear_banned_cpus(self, sysconfig):
		lines = []
		for line in sysconfig.split("\n"):
			if not re.match(r"\s*IRQBALANCE_BANNED_CPUS=", line):
				lines.append(line)
		return "\n".join(lines)

	def _restart_irqbalance(self):
		# Exit code 5 means unit not found (see 'EXIT_NOTINSTALLED' in
		# systemd.exec(5))
		retcode, out = self._cmd.execute(
			["systemctl", "try-restart", "irqbalance"],
			no_errors=[5])
		if retcode != 0:
			log.warn("Failed to restart irqbalance. Is it installed?")

	def _set_banned_cpus(self, banned_cpumask):
		content = self._read_irqbalance_sysconfig()
		if content is None:
			return
		content = self._clear_banned_cpus(content)
		content = self._write_banned_cpus(content, banned_cpumask)
		if self._write_irqbalance_sysconfig(content):
			self._restart_irqbalance()

	def _restore_banned_cpus(self):
		content = self._read_irqbalance_sysconfig()
		if content is None:
			return
		content = self._clear_banned_cpus(content)
		if self._write_irqbalance_sysconfig(content):
			self._restart_irqbalance()

	@command_custom("banned_cpus", per_device=False)
	def _banned_cpus(self, enabling, value, verify, ignore_missing):
		banned_cpumask = None
		if value is not None:
			banned = set(self._cmd.cpulist_unpack(value))
			present = set(self._cpus)
			if banned.issubset(present):
				banned_cpumask = self._cmd.cpulist2hex(list(banned))
			else:
				str_cpus = ",".join([str(x) for x in self._cpus])
				log.error("Invalid banned_cpus specified, '%s' does not match available cores '%s'"
					  % (value, str_cpus))

		if (enabling or verify) and banned_cpumask is None:
			return None
		if verify:
			# Verification is currently not supported
			return None
		elif enabling:
			self._set_banned_cpus(banned_cpumask)
		else:
			self._restore_banned_cpus()

Name	Size	Perms	Modified	Actions
instance DIR	-	drwxr-xr-x	2025-03-30 04:15:50	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2025-03-30 04:15:50	Edit Download recursive recursive
base.py	21.89 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
decorators.py	983 B	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
exceptions.py	99 B	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
hotplug.py	3.84 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_acpi.py	2.39 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_audio.py	3.13 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_bootloader.py	25.31 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_cpu.py	27.56 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_disk.py	16.65 KB	lrw-r--r--	2025-03-11 07:42:24	Edit Download recursive recursive
plugin_eeepc_she.py	2.88 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_irqbalance.py	3.47 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_modules.py	4.81 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_mounts.py	5.45 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_net.py	22.71 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_rtentsk.py	1.08 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_scheduler.py	54.94 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_script.py	3.76 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_scsi_host.py	3.08 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_selinux.py	2.27 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_service.py	10.47 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_sysctl.py	6.74 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_sysfs.py	2.63 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_systemd.py	5.30 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_uncore.py	4.61 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_usb.py	1.97 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_video.py	3.72 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
plugin_vm.py	3.48 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
repository.py	1.49 KB	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive
__init__.py	49 B	lrw-r--r--	2024-02-22 12:23:28	Edit Download recursive recursive

Directory Contents