Current File : //usr/lib/python3.9/site-packages/tuned/daemon/__pycache__/controller.cpython-39.opt-1.pyc
a
���h�J � @ s� d dl mZ d dlZ d dlZ d dlmZ d dlZd dlmZ d dlm Z d dl
m Z d dl Z d dl
Z
dgZe j�� ZG dd� de�ZG d d� de jjj�ZdS )
� )�exportsN)�TunedException)�commands)�hotplug�
Controllerc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TimerStorec C s t � | _t�� | _d S �N)�dict�_timers� threading�Lock� _timers_lock��self� r �;/usr/lib/python3.9/site-packages/tuned/daemon/controller.py�__init__ s zTimerStore.__init__c C s4 | j � || j|< W d � n1 s&0 Y d S r )r
r
�r �token�timerr r r � store_timer s zTimerStore.store_timerc C sV | j �< z | j| }|�� | j|= W n Y n0 W d � n1 sH0 Y d S r )r
r
�cancelr r r r �
drop_timer s
zTimerStore.drop_timerc C sL | j �2 | j�� D ] }|�� q| j�� W d � n1 s>0 Y d S r )r
r
�valuesr �clear)r r r r r �
cancel_all" s
zTimerStore.cancel_allN)�__name__�
__module__� __qualname__r r r r r r r r r s r c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Ze� d
�d d � �Z
d
d� Z e� dd�dZdd��Z
e� dd�d[dd��Ze� dd�d\dd��Zd]dd �Ze� dd�d^d d ��Ze� dd�d_d d ��Zd!d"� Ze� dd#�d`d$d%��Ze� dd#�dad&d'��Ze� dd�dbd(d)��Ze� dd*�dcd+d,��Ze� dd�ddd-d.��Ze� dd�ded/d0��Ze� dd�dfd1d2��Ze� dd3�dgd4d5��Ze� dd6�dhd7d8��Z e� dd9�did:d;��Z e� dd�djd<d=��Z e� dd�dkd>d?��Z e� dd�dld@dA��Z e� ddB�dmdCdD��Z!e� dd�dndEdF��Z"e� ddG�dodHdI��Z#e� dd�dpdJdK��Z$e� dLd#�dqdMdN��Z%e� ddO�drdPdQ��Z&e� ddR�dsdSdT��Z'e� dUd#�dtdVdW��Z(e� dd#�dudXdY��Z)� Z*S )vr z�
Controller's purpose is to keep the program running, start/stop the tuning,
and export the controller interface (currently only over D-Bus).
c s8 t t| ��� || _|| _t�� | _t� | _ t
� | _ d S r ) �superr r �_daemon�_global_configr �Event�
_terminater �_cmdr � _timer_store)r �daemonZ
global_config�� __class__r r r . s
zController.__init__c
C s� t �d� | j�� | j�tjtj�}|dk�r"t �d� t j
� t � � �}| j�
dtj�}z|�|� W n ty� t �d� Y n0 d}t�� }zTt�� || k r�| j�� s�|r�|jdd�}q�| j�� s�|r�t �d � n
t �d
� W n8 ttf�y } zt �d | � W Y d }~n
d }~0 0 ~| �� }| j�tjtj�}|�sP|�rPt�� |�rt| j � | jd��stt� � �qVt �d
� | � � d S )z1
Controller main loop. The call is blocking.
zstarting controllerr zwaiting for udev to settle�udev_buffer_sizez�cannot set udev monitor receive buffer size, we are probably running inside container or with limited capabilites, TuneD functionality may be limitedT� )�timeoutzudev settle timed outz udev settledzudev settle failed, '%s'Nzterminating controller) �log�infor# r r! Zget_int�constsZ CFG_STARTUP_UDEV_SETTLE_WAITZ CFG_DEF_STARTUP_UDEV_SETTLE_WAIT�pyudevZMonitorZ from_netlinkZContextZget_sizeZCFG_DEF_UDEV_BUFFER_SIZEZset_receive_buffer_size�EnvironmentError�warning�time�is_setZpoll�OSError�IOError�start�get_bool�
CFG_DAEMON�CFG_DEF_DAEMONr r$ �waitZ period_check�stop) r Z wait_settleZmonitorr) �p�t�e�resr& r r r �run6 sD
�
$
zController.runc C s | j �� d S r )r# �setr r r r � terminatec s zController.terminatec C s` | j j�� r| j j�� nBd}|r\| j j�� | j j�� | �� sN| j j�� | j j�� }q d S )NT)r Z_sighup_processingr3 Z_sighup_pendingrA r �reload)r Z do_reloadr r r �sighupf s zController.sighupZsbsc C s d S r r )r � profile_name�resultZerrstrr r r �profile_changedr s zController.profile_changedc C s t j�|� | j�|� d S r ��tuned�logs�log_capture_finishr% r )r r r r r �_log_capture_abort{ s z Controller._log_capture_abort�ii�sNc C sf |dkr dS t j�|�}|d u r$dS |dkrVtj|| j|gd�}| j�||� |�� |d u rbdS |S )N� r )�args) rI rJ �log_capture_startr �TimerrL r% r r6 )r Z log_levelr+ �callerr r r r r rQ s �z Controller.log_capture_startc C s4 |dkr dS t j�|�}| j�|� |d u r0dS |S �NrO rH )r r rS r? r r r rK � s
z Controller.log_capture_finishrO �bc C sD |dkr dS | j �tjtj�r:| j�� r,dS | j�� s:dS | j�� S )NrO FT) r! r7 r. r8 r9 r �
is_running�
is_enabledr6 �r rS r r r r6 � s
zController.startFc C s, | j �� sd}n| j j|d�}| j�� |S )NT��profile_switch)r rV r; r% r )r rZ r? r r r �_stop� s
zController._stopc C s |dkr dS | j dd�S )NrO FrY )r[ rX r r r r; � s zController.stopc
C sv |dkr dS | j �� r*| jdd�}|s*dS z| j �� W n4 tyl } z t�d| � W Y d }~dS d }~0 0 | �� S )NrO FTrY zFailed to reload TuneD: %s)r rV r[ Zreload_profile_configr r, �errorr6 )r rS Zstop_okr> r r r rC � s
zController.reloadc
C s | j �� }d}d}d}z�z$|r,| j jdd� | j �||� W nh tjjy� } zLd}t|�}|r�| j jj |kr�t
� d| � d}nt
� d| � W Y d }~n
d }~0 0 W |r�|r�t
� d| � n|s�t
�
d� | j �� n6|�r|r�t
� d| � n|�st
�
d� | j �� 0 ||fS ) N�OKTFrY z@Failed to reapply profile '%s'. Did it change on disk and break?z Failed to apply profile '%s'z>Applying previously applied (possibly out-dated) profile '%s'.z$Applying previously applied profile.)r rV r; Z set_profilerI �
exceptionsr �str�profile�namer, r\ r1 r- r6 )r rE �manualZ was_running�msg�successZreapplyr> r r r �_switch_profile� s8
&
�
zController._switch_profilez(bs)c C s( |dkr dS | j �|�s dS | �|d�S )NrO �F� Unauthorized)FzInvalid profile_nameT)r$ �
is_valid_namere �r rE rS r r r �switch_profile� s
zController.switch_profilec C s |dkr dS | � � }| �|d�S )NrO rf F)�recommend_profilere )r rS rE r r r � auto_profile� s zController.auto_profilec C s* |dkr dS | j jd ur"| j jjS dS d S rT )r r` ra rX r r r �active_profile� s
zController.active_profilez(ss)c
C s� |dkr dS | j j}|d u rxz"| j�� \}}|d u r<|d u}W n8 tyv } z d}t|�}||fW Y d }~S d }~0 0 |r�tjntj}|dfS )NrO )�unknownrg rn ) r rb r$ Zget_active_profiler r_ r. ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)r rS rb r` r> �moder\ r r r � profile_mode� s zController.profile_modec C s |dkr dS | j jpdS rT )r �post_loaded_profilerX r r r rq s z Controller.post_loaded_profilec C sB |dkr dS | j �� r | j �� | j �� r>| j jd dd dd� dS )NrO FT)Zsave_instantly)r rV r; rW Zset_all_profilesrX r r r �disable s
�zController.disablec C s |dkr dS | j �� S )NrO F)r rV rX r r r rV s zController.is_running�asc C s |dkr g S | j jj�� S rT )r �profile_loader�profile_locatorZget_known_namesrX r r r �profiles s zController.profilesza(ss)c C s |dkr g S | j jj�� S rT )r rt ru Zget_known_names_summaryrX r r r � profiles2 s zController.profiles2z(bsss)c C sR |dks| j �|�sdS |d u s(|dkr0| �� }t| jjj�|tj tj
gdg��S )NrO )FrO rO rO ) r$ rh rm �tupler rt ru Zget_profile_attrsr. ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTIONri r r r � profile_info% s
zController.profile_infoc C s |dkr dS | j j�� S rT )r Zprofile_recommenderZ recommendrX r r r rk - s z Controller.recommend_profilec C s |dkr dS | j jdd�S )NrO F�Zignore_missing�r �verify_profilerX r r r r| 3 s zController.verify_profilec C s |dkr dS | j jdd�S )NrO FTrz r{ rX r r r � verify_profile_ignore_missing9 s z(Controller.verify_profile_ignore_missingz a{sa{ss}}c C sr |dkr i S i }| j �� D ]R}|j�d�d �dd�d }|�� }i ||< |�� D ]\}}t|�|| |< qRq|S )zuReturn dictionary with accesible plugins
Return:
dictionary -- {plugin_name: {parameter_name: default_value}}
rO �.����_r* )r �get_all_pluginsr �splitZ_get_config_options�itemsr_ )r rS �pluginsZ plugin_class� plugin_nameZ conf_options�key�valr r r r� ? s zController.get_all_pluginsc C s( |dks| j �|�sdS | j�t|��S )z"Return docstring of plugin's classrO )r$ rh r �get_plugin_documentationr_ �r r� rS r r r r� Q s z#Controller.get_plugin_documentationza{ss}c C s( |dks| j �|�si S | j�t|��S )z�Return dictionary with plugin's parameters and their hints
Parameters:
plugin_name -- name of plugin
Return:
dictionary -- {parameter_name: hint}
rO )r$ rh r �get_plugin_hintsr_ r� r r r r� X s
zController.get_plugin_hintsc C sB |dks| j �|�sdS | jjr>| jjjr>| jjj�|� dS dS )z�Allows to dynamically add sockets to send signals to
Parameters:
path -- path to socket to register for sending signals
Return:
bool -- True on success
rO FT)r$ rh r Z _applicationZ_unix_socket_exporterZregister_signal_path)r �pathrS r r r �register_socket_signal_pathf s
z&Controller.register_socket_signal_path�ssc
C s� |dkr dS | j �|�s dS | j �|�s,dS d}| jjjD ]&}|j|kr:t�d|j � d} qbq:|s�d| }t�|� d|fS t |j
t j �s�d |j
j }t�|� d|fS t
| j �|��}t�d
|jt|�f � | jjjD ]�}|j|@ } t| �r�|| 8 }t�d t| �|j|jf � |j
j|j
jk�rbd |j|j
jt| �|j|j
jf }t�|� d|f S |j
�|| � |j
�|| � q�t|��r�d
t|� }t�|� d|fS dS )NrO rf )FzInvalid devices�F�Invalid instance_nameFzFound instance '%s'.T�Instance '%s' not found�>Plugin '%s' does not support hotplugging or dynamic instances.z-Instance '%s' trying to acquire devices '%s'.�8Moving devices '%s' from instance '%s' to instance '%s'.ztTarget instance '%s' is of type '%s', but devices '%s' are currently handled by instance '%s' which is of type '%s'.z2Ignoring devices not handled by any instance '%s'.�Tr] )r$ rh r �
_unit_manager� instancesra r, �debugr\ �
isinstance�pluginr �PluginrA Z devstr2devsr_ �processed_devices�lenr- �_remove_devices_nocheck�_add_devices_nocheck)
r �devices�
instance_namerS �foundZinstance_target�retsZdevs�instance� devs_movingr r r �instance_acquire_devicesz sX
���
z#Controller.instance_acquire_devicesz (bsa(ss))c s� |dkrddg fS | j �� �s(ddg fS � dkr\� | �� �� vr\d� }t�|� d|g fS tdd� | jjj �}� dkr�t� fdd�|�}d d
t
t d d� |��fS ) a Return a list of active instances of a plugin or all active instances
Parameters:
plugin_name -- name of the plugin or an empty string
Return:
bool -- True on success
string -- error message or "OK"
list of string pairs -- [(instance_name, plugin_name)]
rO Frg �Invalid plugin_namezPlugin '%s' does not existc S s | j S r )Zactive�r� r r r �<lambda>� � z*Controller.get_instances.<locals>.<lambda>c s | j j� kS r )r� ra r� �r� r r r� � r� Tr] c S s | j | jj fS r )ra r� r� r r r r� � r� ) r$ rh r� �keysr, r\ �filterr r� r� �list�map)r r� rS r� r� r r� r �
get_instances� s
zController.get_instancesz(bsas)c C sv |dkrddg fS | j �|�s(ddg fS | jjjD ]&}|j|kr2ddtt|j��f S q2d| }t �
|� d|g fS )z�Return a list of devices assigned to an instance
Parameters:
instance_name -- name of the instance
Return:
bool -- True on success
string -- error message or "OK"
list of strings -- device names
rO Frg r� Tr] r� ) r$ rh r r� r� ra �sortedr� r� r, r\ )r r� rS r� r� r r r �instance_get_devices� s
z Controller.instance_get_deviceszssa{ss}c
C s: |dkr dS | j �|�s dS | j �|�s,dS |�� D ]&\}}| j �|�rT| j �|�s4 dS q4dd� | jjjD �}||�� vr�d| }t�|� d |fS || } t | t
j �s�d
| j }t�|� d |fS |�
d d �}
|�
d
d �} |�
dd �} |�
dd �}
t|�
d| jjj��}z4| �|||
| | |
|�}| �|� | jjj�|� W nH t�y� } z.d|t|�f }t�|� d |fW Y d }~S d }~0 0 t�d||f � | �|� | �|� | jjjD ]z}||k�s�|j| k�s�|j|jk�r�q�| �||j �}t |��r�t�dt|�|j |j f � | � ||� | � ||� �q�dS )a Dynamically create a plugin instance
Parameters:
plugin_name -- name of the plugin
instance_name -- name of the new instance
dict of string-string -- options for the new instance
Return:
bool -- True on success
string -- error message or "OK"
rO rf )Fr� r� )FzInvalid optionsc S s i | ] }|j |�qS r �ra )�.0r< r r r �
<dictcomp>� r� z.Controller.instance_create.<locals>.<dictcomp>zPlugin '%s' not foundFr� r� N�devices_udev_regex�
script_pre� script_post�priorityz Error creating instance '%s': %sz,Created dynamic instance '%s' of plugin '%s'r� r� ) r$ rh r� r r� r� r� r, r\ r� r r� ra �pop�intZ_def_instance_priorityZcreate_instanceZinitialize_instancer� �append� Exceptionr_ r- Zassign_free_devicesZinstance_apply_tuningr� r� Z_get_matching_devicesr� r� r� r� )r r� r� �optionsrS r� �valuer� r� r� r� r� r� r� r� r� r> Zother_instancer� r r r �instance_create� sd
�
�
� zController.instance_createc
sH |dkr dS | j �� �s dS z � fdd�| jjjD �d }W n, tyh d� }t�|� d|f Y S 0 |j}t |t
j �s�d |j }t�|� d|fS |j
�� }z4|�||� | jjj�|� |�|� |�|� W nH t�y } z.d
� t|�f }t�|� d|fW Y d }~S d }~0 0 t�d � � |D ]}|�|� �q2d
S )z�Destroy a dynamically created plugin instance
Parameters:
instance_name -- name of the new instance
Return:
bool -- True on success
string -- error message or "OK"
rO rf r� c s g | ]}|j � kr|�qS r r� )r� �i�r� r r �
<listcomp>& r� z/Controller.instance_destroy.<locals>.<listcomp>r r� Fr� z Error deleting instance '%s': %sNzDeleted instance '%s'r� )r$ rh r r� r� �
IndexErrorr, r\ r� r� r r� ra r� �copyr� �removeZinstance_unapply_tuningZdestroy_instancer� r_ r- Z _add_device) r r� rS r� r� r� r� r> Zdevicer r� r �instance_destroy s:
zController.instance_destroy)N)N)N)F)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)+r r r �__doc__r r@ rB rD r �signalrG rL ZexportrQ rK r6 r[ r; rC re rj rl rm rp rq rr rV rv rw ry rk r| r} r� r� r� r� r� r� r� r� r� �
__classcell__r r r' r r ( s� -
+
?
)rI r Z
tuned.logsZtuned.exceptionsr r Z tuned.constsr. Ztuned.utils.commandsr Z
tuned.pluginsr r/ r2 �__all__rJ �getr, �objectr Z
interfacesZExportableInterfacer r r r r �<module> s
Mini Shell