3
�<�e4 � @ sn d dl mZ d dlT ddlZd dl mZ ddlmZ ddlj Z ddl
Z
ddl Z ej j
� 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<
`systemd`::
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 jjtj�s tjdtj ��tt| �j ||� 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.6/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 r �_instance_cleanup. s z SystemdPlugin._instance_cleanupc C s dd iS )N� cpu_affinityr )�clsr r r �_get_config_options1 s z!SystemdPlugin._get_config_optionsc C sB |d k r>t jd| d |t jd�}|d k r>|jdkr>|jd�S d S )Nz^\s*z \s*=\s*(.*)$)�flagsr )�re�search� MULTILINE� lastindex�group)r �conf�key�mor r r � _get_keyval7 s
zSystemdPlugin._get_keyvalc C s~ t jd| d dt|� |t jd�\}}|dk rzy|d dkrF|d7 }W n tk
r\ Y nX ||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 )Nz^\s*z\s*=.*\n� )r )r �subr )r r" r# r r r �_del_keyK s zSystemdPlugin._del_keyc C s, | j jtjd d�}|d kr(tjd� 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_confN s
z'SystemdPlugin._read_systemd_system_confc C sp t jt j }| jj||�s8tjd� | jj|dd� dS | jj|t j�sltjdt 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_filer3 r4 �unlink�rename)r r" Ztmpfiler r r �_write_systemd_system_confU s
z(SystemdPlugin._write_systemd_system_confc C s t jjtj| j�S )N)r r �joinr ZPERSISTENT_STORAGE_DIR�name)r r r r �_get_storage_filenamec s z#SystemdPlugin._get_storage_filenamec C sl | j � }|d k rh| j� }| jj|d dd�}| jj|� |d krN| j|tj�}n| j|tj|�}| j |� d S )NT)r1 r6 )
r5 r= r r2 r8 r0 r �SYSTEMD_CPUAFFINITY_VARr- r: )r r" �fname�cpu_affinity_savedr r r �_remove_systemd_tuningf s z$SystemdPlugin._remove_systemd_tuningc C s0 |t jkr,tjdt j � | j� tjd� 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_FULLr3 �infor> rA Zconsole)r r Zrollbackr r r �_instance_unapply_staticr s
z&SystemdPlugin._instance_unapply_staticc
C s< |d kr dS dj dd� | jjtjddtjdd|���D ��S )Nr. � c s s | ]}t |�V qd S )N)r* )�.0�vr r r � <genexpr>| s z8SystemdPlugin._cpulist_convert_unpack.<locals>.<genexpr>z\s+�,z,\s+)r; r �cpulist_unpackr r/ )r Zcpulistr r r �_cpulist_convert_unpacky s z%SystemdPlugin._cpulist_convert_unpackr F)Z
per_devicec C s� d }d }| j j| jj| j j|���}djdd� | j j|�D ��}| j� } | d k rh| j| t j
�}| j |�}|r|| j d|||�S |r�| j
� }
| j j|
d dd�} |d k r�| d kr�||kr�| j j|
|dd� tjdt j
|t jf � | j| j| t j
|�� d S ) NrD c s s | ]}t |�V qd S )N)r* )rE rF r r r rG � s z)SystemdPlugin._cmdline.<locals>.<genexpr>r T)r1 r6 )Zmakedirz setting '%s' to '%s' in the '%s')r ZunescapeZ
_variables�expandZunquoter; rI r5 r% r r> rJ Z
_verify_valuer= r2 r7 r3 rB r r: r- ) r Zenabling�valueZverifyZignore_missingZ
conf_affinityZconf_affinity_unpackedrF Z
v_unpackedr" r? r@ r r r �_cmdline~ s"
zSystemdPlugin._cmdline)�__name__�
__module__� __qualname__�__doc__r r r � classmethodr r% r- r0 r5 r: r= rA r Z
ROLLBACK_SOFTrC rJ Zcommand_customrM �
__classcell__r r )r r r
s r )r. r Z
decoratorsZ
tuned.logsZtunedr Ztuned.utils.commandsr Z tuned.constsr r r Zlogs�getr3 ZPluginr r r r r �<module> s