shell bypass 403
3
`��g�4 � @ s� d dl mZ d dlZ d dlZ d dlmZ d dlZd dljZd dlm Z dgZ
e j j � Z
G dd� de�ZG dd� de jjj�ZdS ) � )�exportsN)�TunedException)�commands�
Controllerc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TimerStorec C s t � | _tj� | _d S )N)�dict�_timers� threadingZLock� _timers_lock)�self� r � /usr/lib/python3.6/controller.py�__init__ s zTimerStore.__init__c
C s | j � || j|< W d Q R X d S )N)r
r )r �token�timerr r r
� store_timer s zTimerStore.store_timerc C sB | j �2 y | j| }|j� | j|= W n Y nX W d Q R X d S )N)r
r �cancel)r r r r r r
�
drop_timer s
zTimerStore.drop_timerc C s< | j �, x| jj� D ] }|j� qW | jj� W d Q R X d S )N)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 � Zej d
�d d � �Z
d
d� Z ej dd�dUdd��Z
ej dd�dVdd��Zej dd�dWdd��ZdXdd �Zej dd�dYd d ��Zej dd�dZd d ��Zd!d"� Zej dd#�d[d$d%��Zej dd#�d\d&d'��Zej dd�d]d(d)��Zej dd*�d^d+d,��Zej dd�d_d-d.��Zej dd�d`d/d0��Zej dd�dad1d2��Zej dd3�dbd4d5��Zej dd6�dcd7d8��Z ej dd9�ddd:d;��Z ej dd�ded<d=��Z ej dd�dfd>d?��Z ej dd�dgd@dA��Z ej ddB�dhdCdD��Z!ej dd�didEdF��Z"ej ddG�djdHdI��Z#ej dd�dkdJdK��Z$ej dLd#�dldMdN��Z%ej ddO�dmdPdQ��Z&ej ddR�dndSdT��Z'� Z(S )or 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| �j� || _|| _tj� | _t� | _ t
� | _ d S )N) �superr r �_daemon�_global_configr ZEvent�
_terminater �_cmdr � _timer_store)r �daemonZ
global_config)� __class__r r
r + s
zController.__init__c C sx t jd� | j� }| jjtjtj�}| r6|r6tj� |rb| j j
� x | j j | j d�s`tj
� qFW t jd� | j� 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
�run3 s
zController.runc C s | j j� d S )N)r �set)r r r r
� terminateF s zController.terminatec C s0 | j jj� s,| j jj� | j� s,| j jj� d S )N)r Z_sighup_processingZis_setr. �reloadr )r r r r
�sighupI s zController.sighupZsbsc C s d S )Nr )r � profile_name�resultZerrstrr r r
�profile_changedO s zController.profile_changedc C s t jj|� | jj|� d S )N)�tuned�logs�log_capture_finishr r )r r r r r
�_log_capture_abortX s z Controller._log_capture_abortZii�sNc C sf |dkr dS t jj|�}|d kr$dS |dkrVtj|| j|gd�}| jj||� |j� |d krbdS |S )N� r )�args) r5 r6 �log_capture_startr ZTimerr8 r r r% )r Z log_levelZtimeout�callerr r r r r
r< \ s z Controller.log_capture_startc C s4 |dkr dS t jj|�}| jj|� |d kr0dS |S )Nr: )r5 r6 r7 r r )r r r= r, r r r
r7 j s
z Controller.log_capture_finishr: �bc C sD |dkr dS | j jtjtj�r:| jj� r,dS | jj� s:dS | jj� S )Nr: FT) r r& r' r( r) r �
is_running�
is_enabledr% )r r= r r r
r% r s
zController.startFc C s, | j j� sd}n| j j|d�}| jj� |S )NT)�profile_switch)r r? r+ r r )r rA r, r r r
�_stop} s
zController._stopc C s |dkr dS | j dd�S )Nr: F)rA )rB )r r= r r r
r+ � s zController.stopc C sp |dkr dS | j j� r*| jdd�}|s*dS y| j j� W n. tk
rf } ztjd| � dS d }~X nX | j� S )Nr: FT)rA zFailed to reload TuneD: %s)r r? rB Zreload_profile_configr r# �errorr% )r r= Zstop_ok�er r r
r0 � s
zController.reloadc C s� | j j� }d}d}d}z�y$|r,| j jdd� | j j||� W nf tjjk
r� } zFd}t|�}|r�| j jj |kr�t
j d| � d}nt
j d| � W Y d d }~X nX W d |r�|r�t
j d| � n|s�t
j
d� | j j� X ||fS ) N�OKTF)rA 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 r? r+ Z set_profiler5 �
exceptionsr �str�profile�namer# rC �warnr$ r% )r r2 �manualZ was_running�msg�successZreapplyrD r r r
�_switch_profile� s,
$
zController._switch_profilez(bs)c C s( |dkr dS | j j|�s dS | j|d�S )Nr: F� Unauthorized�Invalid profile_nameT)FrO )FrP )r �
is_valid_namerN )r r2 r= r r r
�switch_profile� s
zController.switch_profilec C s |dkr dS | j � }| j|d�S )Nr: FrO )FrO )�recommend_profilerN )r r= r2 r r r
� auto_profile� s zController.auto_profilec C s* |dkr dS | j jd k r"| j jjS dS d S )Nr: )r rH rI )r r= r r r
�active_profile� s
zController.active_profilez(ss)c C s� |dkr dS | j j}|d krpy"| jj� \}}|d kr<|d k }W n0 tk
rn } zd}t|�}||fS d }~X nX |rztjntj}|dfS )Nr: �unknownrO )rV rO ) r rK r Zget_active_profiler rG r' ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)r r= rK rH rD �moderC r r r
� profile_mode� s zController.profile_modec C s |dkr dS | j jpdS )Nr: )r �post_loaded_profile)r r= r r r
rY � s z Controller.post_loaded_profilec C sB |dkr dS | j j� r | j j� | j j� r>| j jd dd dd� dS )Nr: FT)Zsave_instantly)r r? r+ r@ Zset_all_profiles)r r= r r r
�disable� s
zController.disablec C s |dkr dS | j j� S )Nr: F)r r? )r r= r r r
r? � s zController.is_running�asc C s |dkr g S | j jjj� S )Nr: )r �profile_loader�profile_locatorZget_known_names)r r= r r r
�profiles� s zController.profilesza(ss)c C s |dkr g S | j jjj� S )Nr: )r r\ r] Zget_known_names_summary)r r= r r r
� profiles2� s zController.profiles2z(bsss)c C sT |dks| j j|� rdS |d ks*|dkr2| j� }t| jjjj|tj tj
gdg��S )Nr: F)Fr: r: r: ) r rQ rU �tupler r\ r] Zget_profile_attrsr' ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTION)r r2 r= r r r
� profile_info s
zController.profile_infoc C s |dkr dS | j jj� S )Nr: )r Zprofile_recommenderZ recommend)r r= r r r
rS
s z Controller.recommend_profilec C s |dkr dS | j jdd�S )Nr: F)�ignore_missing)r �verify_profile)r r= r r r
rc s zController.verify_profilec C s |dkr dS | j jdd�S )Nr: FT)rb )r rc )r r= r r r
� verify_profile_ignore_missing s z(Controller.verify_profile_ignore_missingz a{sa{ss}}c C sz |dkr i S i }xd| j j� D ]V}|jjd�d jdd�d }|j� }i ||< x$|j� D ]\}}t|�|| |< qVW q W |S )zuReturn dictionary with accesible plugins
Return:
dictionary -- {plugin_name: {parameter_name: default_value}}
r: �.r" �_���)r �get_all_pluginsr �splitZ_get_config_options�itemsrG )r r= ZpluginsZ plugin_class� plugin_nameZ conf_options�key�valr r r
rh s zController.get_all_pluginsc C s* |dks| j j|� rdS | jjt|��S )z"Return docstring of plugin's classr: )r rQ r �get_plugin_documentationrG )r rk r= r r r
rn . s z#Controller.get_plugin_documentationza{ss}c C s* |dks| j j|� ri S | jjt|��S )z�Return dictionary with plugin's parameters and their hints
Parameters:
plugin_name -- name of plugin
Return:
dictionary -- {parameter_name: hint}
r: )r rQ r �get_plugin_hintsrG )r rk r= r r r
ro 5 s
zController.get_plugin_hintsc C sD |dks| j j|� rdS | jjr@| jjjr@| jjjj|� 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
r: FT)r rQ r Z _applicationZ_unix_socket_exporterZregister_signal_path)r �pathr= r r r
�register_socket_signal_pathC s
z&Controller.register_socket_signal_pathZssc
C s� |dkr dS | j j|�s dS | j j|�s,dS d}x2| jjjD ]$}|j|kr<tjd|j � d}P q<W |s�d| }tj|� d|fS t | j j
|��}tjd |jt |�f � x�| jjjD ]�}|j |@ } t
| �r�|| 8 }tjd
t | �|j|jf � |jj|jjk�r6d |j|jjt | �|j|jjf }tj|� d|fS |jj|| � |jj|| � q�W t
|��r~d t |� }tj|� d|fS dS )Nr: FrO �Invalid devices�Invalid instance_namezFound instance '%s'.TzInstance '%s' not foundz-Instance '%s' trying to acquire devices '%s'.z8Moving 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'.rE )FrO )Frr )Frs )TrE )r rQ r �
_unit_manager� instancesrI r# �debugrC r. Z devstr2devsrG �processed_devices�lenr$ �pluginZ_remove_devices_nocheckZ_add_devices_nocheck)
r Zdevices�
instance_namer= �foundZinstance_target�retsZdevs�instanceZ devs_movingr r r
�instance_acquire_devicesW sJ
z#Controller.instance_acquire_devicesz (bsa(ss))c s� |dkrddg fS | j j� �s(ddg fS � dkr\� | j� j� kr\d� }tj|� 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)]
r: FrO zInvalid plugin_namezPlugin '%s' does not existc S s | j S )N)Zactive)r} r r r
�<lambda>� s z*Controller.get_instances.<locals>.<lambda>c s | j j� kS )N)ry rI )r} )rk r r
r � s TrE c S s | j | jj fS )N)rI ry )r} r r r
r � s ) r rQ rh �keysr# rC �filterr rt ru �list�map)r rk r= r| ru r )rk r
�
get_instances s
zController.get_instancesz(bsas)c C sv |dkrddg fS | j j|�s(ddg fS x0| jjjD ]"}|j|kr4ddtt|j��fS q4W d| }t j
|� 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
r: FrO zInvalid instance_nameTrE zInstance '%s' not found) r rQ r rt ru rI �sortedr� rw r# rC )r rz r= r} r| r r r
�instance_get_devices� s
z Controller.instance_get_devices)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))r r r �__doc__r r- r/ r1 r �signalr4 r8 Zexportr< r7 r% rB r+ r0 rN rR rT rU rX rY rZ r? r^ r_ ra rS rc rd rh rn ro rq r~ r� r� �
__classcell__r r )r! r
r % sv
'
)r5 r Z
tuned.logsZtuned.exceptionsr r Z tuned.constsr' Ztuned.utils.commandsr �__all__r6 �getr# �objectr Z
interfacesZExportableInterfacer r r r r
�<module> s