File: //lib/python3.9/site-packages/tuned/plugins/__pycache__/plugin_systemd.cpython-39.pyc
a
,�gG � @ sp d dl mZ d dlT ddlZd dl mZ ddlmZ ddlm Z ddl
Z
ddl Z ej �
� ZG dd� dej�ZdS ) � )�base)�*� N)�
exceptions)�commandsc s� e Zd ZdZ� fdd�Zdd� Zdd� Zedd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zejfdd�Zdd� Zed d d �d d � �Z� ZS )!�
SystemdPlugina'
Plug-in for tuning systemd options.
The [option]`cpu_affinity` option allows setting CPUAffinity in
`/etc/systemd/system.conf`. This configures the CPU affinity for the
service manager as well as the default CPU affinity for all forked
off processes. The option takes a comma-separated list of CPUs with
optional CPU ranges specified by the minus sign (`-`).
.Set the CPUAffinity for `systemd` to `0 1 2 3`
====
----
[systemd]
cpu_affinity=0-3
----
====
NOTE: These tunings are unloaded only on profile change followed by a reboot.
c s@ t j�tj�s t�dtj ��tt| �j |i |�� t
� | _ d S )NzERequired systemd '%s' configuration file not found, disabling plugin.) �os�path�isfile�consts�SYSTEMD_SYSTEM_CONF_FILEr ZNotSupportedPluginException�superr �__init__r �_cmd)�self�args�kwargs�� __class__� �@/usr/lib/python3.9/site-packages/tuned/plugins/plugin_systemd.pyr " s zSystemdPlugin.__init__c C s d|_ d|_d S )NFT)Z_has_dynamic_tuningZ_has_static_tuning�r �instancer r r �_instance_init( s z SystemdPlugin._instance_initc C s d S �Nr r r r r �_instance_cleanup, s z SystemdPlugin._instance_cleanupc C s dd iS )N� cpu_affinityr )�clsr r r �_get_config_options/ s �z!SystemdPlugin._get_config_optionsc C sB |d ur>t jd| d |t jd�}|d ur>|jdkr>|�d�S d S )N�^\s*z \s*=\s*(.*)$��flagsr )�re�search� MULTILINE� lastindex�group)r �conf�key�mor r r � _get_keyval5 s
zSystemdPlugin._get_keyvalc C s| t jd| d dt|� |t jd�\}}|dk rxz|d dkrF|d7 }W n tyZ Y n0 ||d t|� d 7 }|S |S ) Nz^(\s*z\s*=).*$z\g<1>r r ����
�=)r" �subn�strr$ �
IndexError)r r' r( �valZconf_newZnsubsr r r � _add_keyval= s ( zSystemdPlugin._add_keyvalc C s t jd| d d|t jd�S )Nr z\s*=.*\n� r )r" �subr$ )r r' r( r r r �_del_keyI s zSystemdPlugin._del_keyc C s, | j jtjd d�}|d u r(t�d� d S |S )N)�err_retz(error reading systemd configuration file)r � read_filer r �log�error)r Zsystemd_system_confr r r �_read_systemd_system_confL s
z'SystemdPlugin._read_systemd_system_confc C sp t jt j }| j�||�s8t�d� | jj|dd� dS | j�|t j�slt�dt j � | jj|dd� dS dS )Nz(error writing systemd configuration fileT)�no_errorFz/error replacing systemd configuration file '%s') r r ZTMP_FILE_SUFFIXr �
write_to_filer8 r9 �unlink�rename)r r' Ztmpfiler r r �_write_systemd_system_confS s
z(SystemdPlugin._write_systemd_system_confc C s t j�tj| j�S r )r r �joinr ZPERSISTENT_STORAGE_DIR�name)r r r r �_get_storage_filenamea s z#SystemdPlugin._get_storage_filenamec C sl | � � }|d urh| �� }| jj|d dd�}| j�|� |d u rN| �|tj�}n| �|tj|�}| � |� d S )NT�r6 r; )
r: rB r r7 r= r5 r �SYSTEMD_CPUAFFINITY_VARr2 r? )r r' �fname�cpu_affinity_savedr r r �_remove_systemd_tuningd s z$SystemdPlugin._remove_systemd_tuningc C s0 |t jkr,t�dt j � | �� t�d� d S )Nz6removing '%s' systemd tuning previously added by TuneDz[you may need to manualy run 'dracut -f' to update the systemd configuration in initrd image)r Z
ROLLBACK_FULLr8 �inforD rG Zconsole)r r Zrollbackr r r �_instance_unapply_staticp s
z&SystemdPlugin._instance_unapply_staticc C s< |d u r dS d� dd� | j�t�ddt�dd|���D ��S )Nr3 � c s s | ]}t |�V qd S r �r/ ��.0�vr r r � <genexpr>z � z8SystemdPlugin._cpulist_convert_unpack.<locals>.<genexpr>z\s+�,z,\s+)r@ r �cpulist_unpackr" r4 )r Zcpulistr r r �_cpulist_convert_unpackw s z%SystemdPlugin._cpulist_convert_unpackr F)Z
per_devicec
C s� d }d }| j �| j�| j �|���}d�dd� | j �|�D ��} | �� }
|
d urh| �|
t j
�}| � |�}|r|| � d| ||�S |r�| �
� } | j j| d dd�} |d ur�| d u r�| |kr�| j j| |dd� t�dt j
| t jf � | �| �|
t j
| �� dS d S ) NrJ c s s | ]}t |�V qd S r rK rL r r r rO � rP z)SystemdPlugin._cmdline.<locals>.<genexpr>r TrC )Zmakedirz setting '%s' to '%s' in the '%s')r ZunescapeZ
_variables�expand�unquoter@ rR r: r* r rD rS Z
_verify_valuerB r7 r< r8 rH r r? r2 )
r Zenabling�valueZverifyZignore_missingr Z
conf_affinityZconf_affinity_unpackedrN Z
v_unpackedr' rE rF r r r �_cmdline| s&
zSystemdPlugin._cmdline)�__name__�
__module__� __qualname__�__doc__r r r � classmethodr r* r2 r5 r: r? rB rG r Z
ROLLBACK_SOFTrI rS Zcommand_customrW �
__classcell__r r r r r
s"
r )r3 r Z
decoratorsZ
tuned.logsZtunedr Ztuned.utils.commandsr Z tuned.constsr r r" Zlogs�getr8 ZPluginr r r r r �<module> s