File: //lib/python3.9/site-packages/tuned/plugins/__pycache__/base.cpython-39.pyc
a
Y��dW � @ sn d dl Z d dlmZ d dlZd dlZd dlZd dlmZ d dl Z d dl
m Z m Z ej
�� ZG dd� de�ZdS )� N)�commands)�Popen�PIPEc @ s" e Zd ZdZdd� Zdd� Zdd� Zedd � �Ze d
d � �Z
e d d
� �Z e dd� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z e j fd2d3�Z d4d5� Z!d6d7� Z"d8d9� Z#e j fd:d;�Z$d<d=� Z%d>d?� Z&e j fd@dA�Z'dBdC� Z(dDdE� Z)dFdG� Z*dHdI� Z+dJdK� Z,dLdM� Z-d|dOdP�Z.d}dQdR�Z/d~dSdT�Z0ddUdV�Z1dWdX� Z2dYdZ� Z3d[d\� Z4d]d^� Z5d_d`� Z6d�dbdc�Z7d�ddde�Z8dfdg� Z9dhdi� Z:djdk� Z;d�dldm�Z<d�dndo�Z=dpdq� Z>drds� Z?dtdu� Z@dvdw� ZAdxdy� ZBdzd{� ZCdNS )��Plugina
Base class for all plugins.
Plugins change various system settings in order to get desired performance or power
saving. Plugins use Monitor objects to get information from the running system.
Intentionally a lot of logic is included in the plugin to increase plugin flexibility.
c C sn |� | jj�| _|| _|| _|| _|| _|| _t �
� | _ | � � || _
|| _d| _d| _| �� | _t� | _dS )zPlugin constructor.FN)�create� __class__�__name__�_storageZ_monitors_repositoryZ_hardware_inventory�_device_matcher�_device_matcher_udev�_instance_factory� collections� OrderedDict�
_instances�_init_commands� _global_cfg�
_variables�_has_dynamic_options�_devices_inited�#_get_config_options_used_by_dynamic�_options_used_by_dynamicr �_cmd) �selfZmonitors_repositoryZstorage_factoryZhardware_inventoryZdevice_matcherZdevice_matcher_udevZinstance_factoryZ
global_cfgZ variables� r �6/usr/lib/python3.9/site-packages/tuned/plugins/base.py�__init__ s
zPlugin.__init__c C s | � � d S �N)�destroy_instances�r r r r �cleanup, s zPlugin.cleanupc C s | j s| �� d| _ d S )NT)r �
_init_devicesr r r r � init_devices/ s zPlugin.init_devicesc C s | j j�d�d �dd�d S )N�.����_� )r �
__module__�splitr r r r �name4 s z Plugin.namec C s i S )z-Default configuration options for the plugin.r r r r r �_get_config_options< s zPlugin._get_config_optionsc C s i S )z*Explanation of each config option functionr )�clsr r r �get_config_options_hintsA s z Plugin.get_config_options_hintsc C s g S )znList of config options used by dynamic tuning. Their previous values will be automatically saved and restored.r r r r r r F s z*Plugin._get_config_options_used_by_dynamicc C sL | � � �� }|D ]6}||v s"| jr0|| ||< qt�d|| jjf � q|S )z3Merge provided options with plugin default options.z$Unknown option '%s' for plugin '%s'.)r) �copyr �log�warnr r )r �optionsZ effective�keyr r r �_get_effective_optionsK s z Plugin._get_effective_optionsc C s, t |�tu r|S t|��� }|dkp*|dkS )N�true�1)�type�bool�str�lower)r �valuer r r � _option_boolV s zPlugin._option_boolc C sF || j v rtd| ��| �|�}| j�| ||||||�}|| j |< |S )z8Create new instance of the plugin and seize the devices.z.Plugin instance with name '%s' already exists.)r � Exceptionr1 r r ) r r( �devices_expression�devices_udev_regex�
script_pre� script_postr/ Zeffective_options�instancer r r �create_instance` s
�
zPlugin.create_instancec C sV |j | krtd|| f ��|j| jvr2td| ��| j|j }| �|� | j|j= dS )zDestroy existing instance.z9Plugin instance '%s' does not belong to this plugin '%s'.z+Plugin instance '%s' was already destroyed.N)Z_pluginr: r( r �_destroy_instance�r r? r r r �destroy_instancel s
zPlugin.destroy_instancec C s$ t �d|j| jf � | �|� dS )zInitialize an instance.z initializing instance %s (%s)N)r- �debugr( �_instance_initrB r r r �initialize_instancew s zPlugin.initialize_instancec C sB t | j�� �D ]$}t�d|j| jf � | �|� q| j�� dS )zDestroy all instances.zdestroying instance %s (%s)N)�listr �valuesr- rD r( rA �clearrB r r r r | s zPlugin.destroy_instancesc C s | � |� | �|� d S r )�release_devices�_instance_cleanuprB r r r rA � s
zPlugin._destroy_instancec C s
t � �d S r ��NotImplementedErrorrB r r r rE � s zPlugin._instance_initc C s
t � �d S r rL rB r r r rK � s zPlugin._instance_cleanupc C s d| _ t� | _t� | _d S �NF)�_devices_supported�set�_assigned_devices�
_free_devicesr r r r r � s zPlugin._init_devicesc C s dS )z�Override this in a subclass to transform a list of device names (e.g. ['sda'])
to a list of pyudev.Device objects, if your plugin supports itNr )r �devicesr r r �_get_device_objects� s zPlugin._get_device_objectsc C sj |j d u r t| j�|j|��S | �|�}|d u rDt�d| j� t� S | j �|j |�}tdd� |D ��S d S )Nz<Plugin '%s' does not support the 'devices_udev_regex' optionc S s g | ]
}|j �qS r )Zsys_name)�.0�xr r r �
<listcomp>� � z0Plugin._get_matching_devices.<locals>.<listcomp>)
r< rP r
Z
match_listr; rT r- �errorr( r )r r? rS Z udev_devicesr r r �_get_matching_devices� s
z Plugin._get_matching_devicesc C s� | j s
d S t�d|j � | �|| j�}t|�dk|_|jsNt�d|j � n`|j}|j| jkrn|d| j 7 }t� d|d�
|�f � |j � |� | j
|O _
| j|8 _d S )Nz assigning devices to instance %sr z*instance %s: no matching devices availablez (%s)z!instance %s: assigning devices %s�, )rO r- rD r( rZ rR �len�activer. �info�join�assigned_devices�updaterQ )r r? Z to_assignr( r r r �assign_free_devices� s zPlugin.assign_free_devicesc C sV | j s
d S |j|jB | j@ }d|_|j�� |j�� | j|8 _| j|O _d S rN )rO �processed_devicesr` rQ r] rI rR )r r? Z
to_releaser r r rJ � s ��
zPlugin.release_devicesc C s$ | j s d g}|D ]}|||� qd S r )rO )r r? �callbackrS �devicer r r �_run_for_each_device� s zPlugin._run_for_each_devicec C s d S r r �r r? Zenablingr r r �_instance_pre_static� s zPlugin._instance_pre_staticc C s d S r r rg r r r �_instance_post_static� s z Plugin._instance_post_staticc C sj |d u r d S t |�dkr0t�d|j|f � d S |�d�sHt�d� dS tj�|�}d}|D �]}tj } | �
| j � � � |g}
|t
jkr�|
�d� |
�|� t�d|t|
�f � t�d tt| �� �� � zVt|g|
ttd| |dd
�} | �� \} }
| j�r t�d || j|
d d � f � d}W q\ ttf�yb } z"t�d
||f � d}W Y d }~q\d }~0 0 q\|S )Nr z1Instance '%s': no device to call script '%s' for.�/zQRelative paths cannot be used in script_pre or script_post. Use ${i:PROFILE_DIR}.FTZ
full_rollbackz'calling script '%s' with arguments '%s'zusing environment '%s')�stdout�stderrZ close_fds�env�cwdZuniversal_newlineszscript '%s' error: %d, '%s'r# zscript '%s' error: %s)r\ r- r. r( �
startswithrY �os�path�dirname�environra r Zget_env�constsZ
ROLLBACK_FULL�appendr^ r6 rD rG �itemsr r Z communicate�
returncode�OSError�IOError)r r? Zscript�oprS �rollbackZdir_name�ret�devrs Z arguments�proc�out�err�er r r �_call_device_script� sB
� zPlugin._call_device_scriptc C s� |j s
dS |jrZ| �||jd|j� | �|d� | �|� | �|d� | �||jd|j� |j r�| j
� t j
t j�r�| �|| j|j� |j�|j� |j�� dS )zG
Apply static and dynamic tuning if the plugin instance is active.
NZapplyT)r] �has_static_tuningr� r= r` rh �_instance_apply_staticri r> �has_dynamic_tuningr �getrt �CFG_DYNAMIC_TUNING�CFG_DEF_DYNAMIC_TUNINGrf �_instance_apply_dynamicrc ra rI rB r r r �instance_apply_tuning� s
�
�z Plugin.instance_apply_tuningc C s� |j s
dS t|j�dkr.t�dd�|j� � |j�� }|jr�| � ||j
d|�dkrXdS | � |||�dkrndS | � ||j d|�dkr�dS dS dS dS )z<
Verify static tuning if the plugin instance is active.
Nr z)BUG: Some devices have not been tuned: %sr[ ZverifyFT)
r] r\ r` r- rY r_ rc r, r� r� r= �_instance_verify_staticr> )r r? �ignore_missingrS r r r �instance_verify_tuning s
�
z Plugin.instance_verify_tuningc C s< |j s
dS |jr8| j�tjtj�r8| �|| j|j �
� � dS )z<
Apply dynamic tuning if the plugin instance is active.
N) r] r� r r� rt r� r� rf �_instance_update_dynamicrc r, rB r r r �instance_update_tuning$ s z Plugin.instance_update_tuningc C s� |t jkrdS |jr8| j�t jt j�r8| �|| j|j � |j
r�| j ||j d|j |d� | �
|d� | �||� | �|d� | j ||jd|j |d� dS )z8
Remove all tunings applied by the plugin instance.
NZunapply)r{ F)rt Z
ROLLBACK_NONEr� r r� r� r� rf �_instance_unapply_dynamicrc r� r� r> rh �_instance_unapply_staticri r= �r r? r{ r r r �instance_unapply_tuning- s
� z Plugin.instance_unapply_tuningc C s | � |� | �||j� d S r )� _execute_all_non_device_commands� _execute_all_device_commandsr` rB r r r r� ? s
z Plugin._instance_apply_staticc C s2 d}| � ||�dkrd}| �|||�dkr.d}|S �NTF)� _verify_all_non_device_commands�_verify_all_device_commands)r r? r� rS r| r r r r� C s z Plugin._instance_verify_staticc C s | � ||j� | �|� d S r )� _cleanup_all_device_commandsrc � _cleanup_all_non_device_commandsr� r r r r� K s �z Plugin._instance_unapply_staticc sB � ��fdd��j D �D ]}����j| � � q���� � d S )Nc s( g | ] }�� ��j| � �d u r|�qS r )� _storage_get� _commands)rU �opt�re r? r r r rW Q rX z2Plugin._instance_apply_dynamic.<locals>.<listcomp>)r �_check_and_save_valuer� r� )r r? re �optionr r� r r� P s z Plugin._instance_apply_dynamicc C s
t � �d S r rL �r r? re r r r r� V s z Plugin._instance_unapply_dynamicc C s
t � �d S r rL r� r r r r� Y s z Plugin._instance_update_dynamicc C s t �� | _| �� | �� dS )z
Initialize commands.
N)r
r r� �_autoregister_commands�_check_commandsr r r r r ` s
zPlugin._init_commandsc C s� | j jD ]�}|�d�rqt| |�}t|d�s.q|jd }| j�|d|i�}d|jv r�d|d< ||d< |jd |d< |jd |d< nBd |jv r�||d < n.d|jv r�||d< |jd |d< |jd |d< || j|< qt� t
t | j� � �d
d � d ��| _dS )
zd
Register all commands marked using @command_set, @command_get, and @command_custom decorators.
�__�_commandr( rP N�custom�
per_device�priorityr� c S s | d d S )Nr% r� r )Z name_infor r r �<lambda>� rX z/Plugin._autoregister_commands.<locals>.<lambda>)r0 )
r �__dict__ro �getattr�hasattrr� r� r� r
r �sorted�iterrv )r � member_name�member� command_namer^ r r r r� h s*
z Plugin._autoregister_commandsc C sF t | j�� �D ]2\}}|�dd�r$qd|vs4d|vrtd| ��qdS )z2
Check if all commands are defined correctly.
r� Fr� rP z,Plugin command '%s' is not defined correctlyN)rG r� rv r� � TypeError)r r� �commandr r r r� � s
zPlugin._check_commandsNc C sJ t | �j}|d u rdn|}|d u r&dn|}|d u r6dn|}d||||f S )N� z %s/%s/%s/%s)r4 r )r Z
instance_namer� � device_name�
class_namer r r � _storage_key� s
�zPlugin._storage_keyc C s& | � |j|d |�}| j�||� d S �Nr( )r� r( r rP )r r? r� r8 r� r0 r r r � _storage_set� s zPlugin._storage_setc C s | � |j|d |�}| j�|�S r� )r� r( r r� �r r? r� r� r0 r r r r� � s zPlugin._storage_getc C s | � |j|d |�}| j�|�S r� )r� r( r Zunsetr� r r r �_storage_unset� s zPlugin._storage_unsetc C sR dd� t | j�� �D �D ]4}| j�|j�|d d ��}|d ur| �|||� qd S )Nc S s g | ]}|d s|�qS �r� r �rU r� r r r rW � rX z;Plugin._execute_all_non_device_commands.<locals>.<listcomp>r( )rG r� rH r �expandr/ r� �_execute_non_device_command�r r? r� � new_valuer r r r� � s z'Plugin._execute_all_non_device_commandsc C s` dd� t | j�� �D �D ]B}| j�|j�|d d ��}|d u r@q|D ]}| �||||� qDqd S )Nc S s g | ]}|d r|�qS r� r r� r r r rW � rX z7Plugin._execute_all_device_commands.<locals>.<listcomp>r( )rG r� rH r r� r/ r� �_execute_device_command)r r? rS r� r� re r r r r� � s z#Plugin._execute_all_device_commandsc C s` d}dd� t | j�� �D �D ]>}| j�|j�|d d ��}|d ur | �||||�dkr d}q |S )NTc S s g | ]}|d s|�qS r� r r� r r r rW � rX z:Plugin._verify_all_non_device_commands.<locals>.<listcomp>r( F)rG r� rH r r� r/ r� �_verify_non_device_command)r r? r� r| r� r� r r r r� � s z&Plugin._verify_all_non_device_commandsc C sf d}dd� t | j�� �D �D ]D}|j�|d d �}|d u r<q |D ] }| �|||||�dkr@d}q@q |S )NTc S s g | ]}|d r|�qS r� r r� r r r rW � rX z6Plugin._verify_all_device_commands.<locals>.<listcomp>r( F)rG r� rH r/ r� �_verify_device_command)r r? rS r� r| r� r� re r r r r� � s z"Plugin._verify_all_device_commandsc
C s� |d ur�t |�}t|�dkr |S |d d� }|dd � }|d u rP|dv rL|S |S zN|dkrxt|�t|�krp|W S W d S n$|dkr�t|�t|�k r�|W S W d S W n( ty� t�d||||f � Y n0 |S )Nr% )�<�>r� r� zhcannot compare new value '%s' with current value '%s' by operator '%s', using '%s' directly as new value)r6 r\ �int�
ValueErrorr- r. )r r� �
current_valueZnwsrz �valr r r � _process_assignment_modifiers� s(
z$Plugin._process_assignment_modifiersFc C s& |d ur|d ||d�S |d � S d S )Nr� �r� r )r r� re r� r r r �_get_current_value� s zPlugin._get_current_valuec C s< | � ||�}| �||�}|d ur8|d ur8| �||||� |S r )r� r� r� )r r? r� re r� r� r r r r� � s
z Plugin._check_and_save_valuec C sP |d d ur"|d d||dd� n*| � ||||�}|d urL|d ||dd� d S �Nr� TFrP �Zsim�r� )r r? r� re r� r r r r� � s
z Plugin._execute_device_commandc C sL |d d ur |d d|dd� n(| � ||d |�}|d urH|d |dd� d S r� r� r� r r r r� � s
z"Plugin._execute_non_device_commandc C s. | j �t|��}t�d|�r*t�dd|�S |S )Nz \s*(0+,?)+([\da-fA-F]*,?)*\s*$z ^\s*(0+,?)+r� )r �unquoter6 �re�match�sub)r r8 �vr r r � _norm_value s zPlugin._norm_valuec C s |d u r d S d}|d u rN|rN|d u r6t �tj| � nt �tj||f � dS |d u�r| �|�}| �|�}zt|�t|�k}W n� t�y zt|d�t|d�k}W nV ty� t|�t|�k}|s�t|�� d�}|D ] }|�
� }||k}|r� q�q�Y n0 Y n0 | j |||||d� |S )NFT� �|)re ) r- r^ rt Z STR_VERIFY_PROFILE_VALUE_MISSINGZ'STR_VERIFY_PROFILE_DEVICE_VALUE_MISSINGr� r� r� r6 r' �strip�_log_verification_result) r r( r� r� r� re r| Zvalsr� r r r �
_verify_value
s:
�zPlugin._verify_valuec C s� |rL|d u r*t �tj|t|��� f � n t �tj||t|��� f � dS |d u r|t �tj|t|��� t|��� f � n(t �tj ||t|��� t|��� f � dS d S r� )
r- r^ rt ZSTR_VERIFY_PROFILE_VALUE_OKr6 r� Z"STR_VERIFY_PROFILE_DEVICE_VALUE_OKrY Z STR_VERIFY_PROFILE_VALUE_FAILZ$STR_VERIFY_PROFILE_DEVICE_VALUE_FAIL)r r( �successr� r� re r r r r� - s ((z Plugin._log_verification_resultc C sn |d d ur |d d||d|�S | j |||d�}| �||�}|d u rHd S |d ||d�}| �|d ||||�S )Nr� Tr� rP r( �r� r� r� )r r? r� re r� r� r� r r r r� < s z Plugin._verify_device_commandc C sb |d d ur |d d|d|�S | � |�}| �||�}|d u r@d S |d |d�}| �|d |||�S )Nr� TrP r( r� )r r? r� r� r� r� r r r r� F s
z!Plugin._verify_non_device_commandc C sV t dd� t| j�� �D ��D ]4}|j�|d d �d usD|d | jv r | �||� q d S )Nc S s g | ]}|d s|�qS r� r r� r r r rW Q rX z;Plugin._cleanup_all_non_device_commands.<locals>.<listcomp>r( )�reversedrG r� rH r/ r� r �_cleanup_non_device_command)r r? r� r r r r� P s $z'Plugin._cleanup_all_non_device_commandsc C sb t dd� t| j�� �D ��D ]@}|j�|d d �d usD|d | jv r |D ]}| �|||� qHq d S )Nc S s g | ]}|d r|�qS r� r r� r r r rW V rX z7Plugin._cleanup_all_device_commands.<locals>.<listcomp>r( )r� rG r� rH r/ r� r �_cleanup_device_command)r r? rS r� re r r r r� U s $z#Plugin._cleanup_all_device_commandsc C s\ |d d ur"|d dd |dd� n6| � |||�}|d urJ|d ||dd� | �|||� d S �Nr� FrP r� �r� r� )r r? r� re � old_valuer r r r� [ s z Plugin._cleanup_device_commandc C sT |d d ur |d dd dd� n0| � ||�}|d urD|d |dd� | �||� d S r� r� )r r? r� r� r r r r� d s z"Plugin._cleanup_non_device_command)NNN)N)N)N)NF)NN)N)N)Dr r& � __qualname__�__doc__r r r! �propertyr( � classmethodr) r+ r r1 r9 r@ rC rF r rA rE rK r rT rZ rb rJ rf rh ri rt Z
ROLLBACK_SOFTr� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s�
# �
! �
r )r� Z tuned.constsrt Ztuned.profiles.variablesZtunedZ
tuned.logsr
Ztuned.utils.commandsr rp �
subprocessr r Zlogsr� r- �objectr r r r r �<module> s