관리-도구
편집 파일: 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