File: //usr/lib/python3.9/site-packages/tuned/daemon/__pycache__/controller.cpython-39.pyc
a
,�g�F � @ s� d dl mZ d dlZ d dlZ d dlmZ d dlZd dlmZ d dlm Z d dl
m Z dgZ e 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 sr t �d� | �� }| j�tjtj�}|s4|r4t�� |r\| j �
� | j � | j d�s\t�
� qBt �d� | �� dS )z1
Controller main loop. The call is blocking.
zstarting controller� zterminating controllerN)�log�info�startr! �get_bool�consts�
CFG_DAEMON�CFG_DEF_DAEMONr r# r r$ �waitZ period_check�stop)r �resr& r r r �run4 s
zController.runc C s | j �� d S r )r# �setr r r r � terminateG 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_processing�is_setZ_sighup_pendingr5 r �reload)r Z do_reloadr r r �sighupJ s zController.sighupZsbsc C s d S r r )r � profile_name�resultZerrstrr r r �profile_changedV 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) r> r? �log_capture_startr �TimerrA r% r r, )r Z log_level�timeout�callerr r r r r rF c s �z Controller.log_capture_startc C s4 |dkr dS t j�|�}| j�|� |d u r0dS |S �NrD r= )r r rI r3 r r r r@ q s
z Controller.log_capture_finishrD �bc C sD |dkr dS | j �tjtj�r:| j�� r,dS | j�� s:dS | j�� S )NrD FT) r! r- r. r/ r0 r �
is_running�
is_enabledr, �r rI r r r r, y s
zController.startFc C s, | j �� sd}n| j j|d�}| j�� |S )NT��profile_switch)r rL r2 r% r )r rP r3 r r r �_stop� s
zController._stopc C s |dkr dS | j dd�S )NrD FrO )rQ rN r r r r2 � 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 )NrD FTrO zFailed to reload TuneD: %s)r rL rQ Zreload_profile_configr r* �errorr, )r rI Zstop_ok�er r r r8 � 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�OKTFrO 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 rL r2 Z set_profiler> �
exceptionsr �str�profile�namer* rR Zwarningr+ r, )r r: �manualZ was_running�msg�successZreapplyrS r r r �_switch_profile� s8
&
�
zController._switch_profilez(bs)c C s( |dkr dS | j �|�s dS | �|d�S )NrD �F� Unauthorized)FzInvalid profile_nameT)r$ �
is_valid_namer\ �r r: rI r r r �switch_profile� s
zController.switch_profilec C s |dkr dS | � � }| �|d�S )NrD r] F)�recommend_profiler\ )r rI r: r r r � auto_profile� s zController.auto_profilec C s* |dkr dS | j jd ur"| j jjS dS d S rJ )r rW rX rN 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 )NrD )�unknownr^ re ) r rY r$ Zget_active_profiler rV r. ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)r rI rY rW rS �moderR r r r � profile_mode� s zController.profile_modec C s |dkr dS | j jpdS rJ )r �post_loaded_profilerN r r r rh � s z Controller.post_loaded_profilec C sB |dkr dS | j �� r | j �� | j �� r>| j jd dd dd� dS )NrD FT)Zsave_instantly)r rL r2 rM Zset_all_profilesrN r r r �disable� s
�zController.disablec C s |dkr dS | j �� S )NrD F)r rL rN r r r rL � s zController.is_running�asc C s |dkr g S | j jj�� S rJ )r �profile_loader�profile_locatorZget_known_namesrN r r r �profiles� s zController.profilesza(ss)c C s |dkr g S | j jj�� S rJ )r rk rl Zget_known_names_summaryrN r r r � profiles2 s zController.profiles2z(bsss)c C sR |dks| j �|�sdS |d u s(|dkr0| �� }t| jjj�|tj tj
gdg��S )NrD )FrD rD rD ) r$ r_ rd �tupler rk rl Zget_profile_attrsr. ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTIONr` r r r � profile_info s
zController.profile_infoc C s |dkr dS | j j�� S rJ )r Zprofile_recommenderZ recommendrN r r r rb s z Controller.recommend_profilec C s |dkr dS | j jdd�S )NrD F�Zignore_missing�r �verify_profilerN r r r rs s zController.verify_profilec C s |dkr dS | j jdd�S )NrD FTrq rr rN r r r � verify_profile_ignore_missing 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}}
rD �.����_r) )r �get_all_pluginsr �splitZ_get_config_options�itemsrV )r rI �pluginsZ plugin_class� plugin_nameZ conf_options�key�valr r r rx # s zController.get_all_pluginsc C s( |dks| j �|�sdS | j�t|��S )z"Return docstring of plugin's classrD )r$ r_ r �get_plugin_documentationrV �r r| rI r r r r 5 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}
rD )r$ r_ r �get_plugin_hintsrV r� r r r r� < 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
rD FT)r$ r_ r Z _applicationZ_unix_socket_exporterZregister_signal_path)r �pathrI r r r �register_socket_signal_pathJ 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 )NrD r] )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'.�TrT )r$ r_ r �
_unit_manager� instancesrX r* �debugrR �
isinstance�pluginr �Pluginr5 Z devstr2devsrV �processed_devices�lenr+ �_remove_devices_nocheck�_add_devices_nocheck)
r �devices�
instance_namerI �foundZinstance_target�retsZdevs�instance� devs_movingr r r �instance_acquire_devices^ 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)]
rD Fr^ �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� rX r� �r| r r r� � r� TrT c S s | j | jj fS r )rX r� r� r r r r� � r� ) r$ r_ rx �keysr* rR �filterr r� r� �list�map)r r| rI 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
rD Fr^ r� TrT r� ) r$ r_ r r� r� rX �sortedr� r� r* rR )r r� rI 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"
rD r] )Fr� r� )FzInvalid optionsc S s i | ] }|j |�qS r �rX )�.0�pr 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$ r_ rz r r� r{ r� r* rR r� r r� rX �pop�intZ_def_instance_priorityZcreate_instanceZinitialize_instancer� �append� ExceptionrV r+ Zassign_free_devicesZinstance_apply_tuningr� r� Z_get_matching_devicesr� r� r� r� )r r| r� �optionsrI r} �valuer{ r� r� r� r� r� r� r� r� rS 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"
rD r] 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$ r_ r r� r� �
IndexErrorr* rR r� r� r r� rX r� �copyr� �removeZinstance_unapply_tuningZdestroy_instancer� rV r+ Z _add_device) r r� rI r� r� r� r� rS 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 r4 r6 r9 r �signalr< rA ZexportrF r@ r, rQ r2 r8 r\ ra rc rd rg rh ri rL rm rn rp rb rs rt rx r r� r� r� r� r� r� r� �
__classcell__r r r' r r & s�
+
?
)r> r Z
tuned.logsZtuned.exceptionsr r Z tuned.constsr. Ztuned.utils.commandsr Z
tuned.pluginsr �__all__r? �getr* �objectr Z
interfacesZExportableInterfacer r r r r �<module> s