File manager

3

�<�e�
�@sddd
lm
Z

ddlmZ
ddlmZ
ddlZddlZddlZddl	Z	ej
j�ZGdd�de
j
�ZdS)�
)
�base)
�command_custom�)
�constsNcs�eZ
dZd
Z�f
dd�Zdd�Zdd�Zedd	��
Zd
d�Z	dd
�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zeddd�dd��
Z�ZS)�IrqbalancePlugina�

	`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
	----
	====
	c
s tt
|�j|
|�

tj�|_dS)
N)�superr�__init__�perfZcpu_map�_cpus)�self�args�kwargs)
�	__class__��'/usr/lib/python3.6/plugin_irqbalance.pyr s

zIrqbalancePlugin.__init__cCsd
|
_d|
_
dS)NFT)Z_has_dynamic_tuningZ_has_static_tuning)r�instancerrr�_instance_init$s

zIrqbalancePlugin._instance_initc
CsdS)
Nr)rrrrr�_instance_cleanup(s
z"IrqbalancePlugin._instance_cleanupc

Csd
di
S)N�banned_cpusr)
�clsrrr�_get_config_options+sz$IrqbalancePlugin._get_config_optionsc
Csly tt
jd
��
}
|
j�SQRXWnFtk
rf
}
z*|jtjkrJtjd�

ntj	d|�

dSd}~XnXdS)N�
rz>irqbalance sysconfig file is missing. Is irqbalance installed?z,Failed to read irqbalance sysconfig file: %s)
�openr�IRQBALANCE_SYSCONFIG_FILE�read�IOError�errno�ENOENT�log�warn�error)r�
f�
errr�_read_irqbalance_sysconfig1s






z+IrqbalancePlugin._read_irqbalance_sysconfigcCsZy&tt
jd
��}|j|
�

WdQRXdStk
rT
}
ztjd|�

dSd}~XnXdS)N�
wTz-Failed to write irqbalance sysconfig file: %sF)rrr�writerrr )r�contentr!r"rrr�_write_irqbalance_sysconfig<s






z,IrqbalancePlugin._write_irqbalance_sysconfigcCs|
d
|S)NzIRQBALANCE_BANNED_CPUS=%s
r)r�	sysconfig�banned_cpumaskrrr�_write_banned_cpusEs
z#IrqbalancePlugin._write_banned_cpuscCs8g}x(|
jd
�
D]}t
jd|�s|j|�

qWd
j|�
S)N�

z\s*IRQBALANCE_BANNED_CPUS=)�split�re�match�append�join)rr(�lines�linerrr�_clear_banned_cpusHs





z#IrqbalancePlugin._clear_banned_cpusc
Cs2|jj
d
ddgdg
d�\}
}|
dkr.tjd�

dS)NZ	systemctlztry-restartZ
irqbalance�)
Z	no_errorsrz.Failed to restart irqbalance. Is it installed?)�_cmdZexecuterr)rZretcode�outrrr�_restart_irqbalanceOs




z$IrqbalancePlugin._restart_irqbalancecCs@|j�}|dkrdS|j
|�
}|j||
�}|j|�
r<|j�
dS)
N)r#r3r*r'r7)rr)r&rrr�_set_banned_cpusXs








z!IrqbalancePlugin._set_banned_cpusc
Cs4|j�}
|
dkrdS|j
|
�
}
|j|
�
r0|j�
dS)
N)r#r3r'r7)rr&rrr�_restore_banned_cpusas







z%IrqbalancePlugin._restore_banned_cpusrF)
Z
per_devicec	Cs�d}|dk	rjt|j
j|�
�
}t|j�
}|j|�
rB|j
jt|�
�
}n(d
jdd�|jD�
�
}tj	d||f�

|
sr|r~|dkr~dS|r�dS|
r�|j
|�

n|j�
dS)N�
,c
Ssg|]}
t|
�
�qSr)
�str)�.0�
xrrr�
<listcomp>rsz1IrqbalancePlugin._banned_cpus.<locals>.<listcomp>zGInvalid banned_cpus specified, '%s' does not match available cores '%s')�setr5Zcpulist_unpackr
�issubsetZcpulist2hex�listr0rr r8r9)	rZenabling�valueZverifyZignore_missingr)ZbannedZpresentZstr_cpusrrr�_banned_cpusis 












zIrqbalancePlugin._banned_cpus)�__name__�
__module__�__qualname__�__doc__rrr�classmethodrr#r'r*r3r7r8r9rrC�
__classcell__rr)
rrrs			r)�rZ
decoratorsrZtunedrZ
tuned.logsrr	r-Zlogs�getrZPluginrrrrr�<module>
s