OwlCyberSecurity - MANAGER
Edit File: admin.cpython-36.pyc
3 �<�e�> � @ s� d dl mZ d dlZd dlmZ d dlmZ ddl m Z d dl m Z d dl jZd dlmZ d dlZd dlZd dlZd dlZd dlZd dlZG d d � d e�ZdS ) � )�print_functionN)�commands)�Locator� )�TunedAdminDBusException)�TunedException)�ProfileRecommenderc @ s� e Zd Zdddejejfdd�Zdd� Zdd� Z d d � Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zdfd d!�Zdgd"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)dJdK� Z*dLdM� Z+dNdO� Z,dhdQdR�Z-didSdT�Z.djdUdV�Z/dkdWdX�Z0dYdZ� Z1d[d\� Z2d]d^� Z3d_d`� Z4dadb� Z5dcdd� Z6deS )l�AdminTFc C s� || _ || _|| _|| _t|�| _ttj�| _ t j � | _ d| _ d| _d| _d | _d | _|| _t� | _| j r�tjjtjtjtj|�| _y| jjtj | j � W n2 t k r� } z| j |� d| _ W Y d d }~X nX d S )N� TF) �_dbusZ_debug�_async�_timeoutr �_cmd�profiles_locator�constsZLOAD_DIRECTORIES�_profiles_locator� threadingZEvent�_daemon_action_finished�_daemon_action_profile�_daemon_action_result�_daemon_action_errstr� _controller� _log_token� _log_levelr �_profile_recommender�tunedZadminZDBusControllerZDBUS_BUSZDBUS_INTERFACEZ DBUS_OBJECTZset_signal_handlerZSIGNAL_PROFILE_CHANGED�_signal_profile_changed_cbr �_error)�selfZdbus�debugZasynco�timeoutZ log_level�e� r" �/usr/lib/python3.6/admin.py�__init__ s* zAdmin.__init__c C s t |tjd� d S )N)�file)�print�sys�stderr)r �messager" r" r# r + s z Admin._errorc C s* | j j� s&|| _|| _|| _| j j� d S )N)r Zis_setr r r �set)r � profile_name�resultZerrstrr" r" r# r . s z Admin._signal_profile_changed_cbc C sn y t jt| jjtj��d� W nJ tk rF } z |jtj kS d }~X n$ t t fk rh } zdS d }~X nX dS )Nr FT) �os�kill�intr Z read_filer ZPID_FILE�OSError�errnoZEPERM� ValueError�IOError)r r! r" r" r# �_tuned_is_running6 s zAdmin._tuned_is_runningc - O s6 |d ks|dkrdS d }d }d}yt | d| �}W n( tk rZ } z d| _W Y d d }~X nX yt | d| �}W nB tk r� } z&| js�| jt|�d| � dS W Y d d }~X nX | j�r y0| jj| j� | jj|f|�|� | jj � }W n4 t k �r } z| j|� d| _W Y d d }~X nX | j�s2|||�}|S )Nr FZ _action_dbus_Z_action_z , action '%s' is not implemented) �getattr�AttributeErrorr r �strr Zset_on_exit_action�_log_capture_finish� set_actionZrunr )r Z action_name�args�kwargs�actionZ action_dbus�resr! r" r" r# r< @ s6 z Admin.actionc C sh t d� xZ|D ]R}|d d k rP|d dkrPt | jjd|d dd|d �� qt d|d � qW d S )NzAvailable profiles:r r z- %sr � )r& r Z align_str)r � profile_names�profiler" r" r# �_print_profiles^ s &zAdmin._print_profilesc C sd y| j j� }W n6 tk rD } zdd� | j j� D �}W Y d d }~X nX | j|� | j� | j jd�S )Nc S s g | ] }|d f�qS )r r" )�.0r@ r" r" r# � <listcomp>k s z4Admin._action_dbus_list_profiles.<locals>.<listcomp>T)r Z profiles2r �profilesrA �_action_dbus_active�exit)r r? r! r" r" r# �_action_dbus_list_profilesf s & z Admin._action_dbus_list_profilesc C s | j | jj� � | j� dS )NT)rA r Zget_known_names_summary�_action_active)r r" r" r# �_action_list_profilesp s zAdmin._action_list_profilesc C s& | j j� }|dkrd }| j jd� |S )Nr T)r �active_profilerF )r r+ r" r" r# �_dbus_get_active_profileu s z Admin._dbus_get_active_profilec C s | j j� \}}|S )N)r �get_active_profile)r r+ �manualr" r" r# �_get_active_profile| s zAdmin._get_active_profilec C s. | j j� \}}|d kr |d k }|r(tjS tjS )N)r rL r ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)r r@ rM r" r" r# �_get_profile_mode� s zAdmin._get_profile_modec C s | j j� }|dkrd }|S )Nr )r �post_loaded_profile)r r+ r" r" r# � _dbus_get_post_loaded_profile� s z#Admin._dbus_get_post_loaded_profilec C s | j j� }|S )N)r Zget_post_loaded_profile)r r+ r" r" r# �_get_post_loaded_profile� s z Admin._get_post_loaded_profilec C sl |d dkrXt d� t |d � t � t d� t |d � t � t d� t |d � dS t d | � d S d S ) Nr Tz Profile name:r zProfile summary:� zProfile description:� z,Unable to get information about profile '%s'F)r& )r r@ � profile_infor" r" r# �_print_profile_info� s zAdmin._print_profile_infor c C sB |dkr| j � }|r*| j|| jj|��}n td� d}| jj|�S )Nr zNo current active profile.F)rK rV r rU r& rF )r r@ r= r" r" r# �_action_dbus_profile_info� s z Admin._action_dbus_profile_infoc C s| |dkrXy | j � }|d kr&td� dS W n. tk rV } z| jt|�� dS d }~X nX | j|| jj|tj tj gddg��S )Nr zNo current active profile.F) rN r& r r r7 rV r Zget_profile_attrsr ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTION)r r@ r! r" r" r# �_action_profile_info� s zAdmin._action_profile_infoc C s$ |d krt d� dS t d| � dS )NzNo current active profile.FzCurrent active profile: %sT)r& )r r+ r" r" r# �_print_profile_name� s zAdmin._print_profile_namec C s |rt d| � d S )Nz Current post-loaded profile: %s)r& )r r+ r" r" r# �_print_post_loaded_profile� s z Admin._print_post_loaded_profilec C s4 | j � }| j|�}|r(| j� }| j|� | jj|�S )N)rK rY rQ rZ r rF )r rJ r= rP r" r" r# rE � s zAdmin._action_dbus_activec C s� y2| j � }| j� }|r0|r$|d7 }nd}||7 }W n. tk r` } z| jt|�� dS d }~X nX |d k r�| j� r�td� td| � |r�td| � dS | j|�}| j|� |S )N� r FzKIt seems that tuned daemon is not running, preset profile is not activated.zPreset profile: %sz Preset post-loaded profile: %sT) rN rR r r r7 r4 r& rY rZ )r r+ rP r! r= r" r" r# rH � s( zAdmin._action_activec C s t d| � d S )NzProfile selection mode: )r& )r �moder" r" r# �_print_profile_mode� s zAdmin._print_profile_modec C sB | j j� \}}| j|� |dkr6| j|� | j jd�S | j jd�S )Nr FT)r Z profile_moder] r rF )r r\ �errorr" r" r# �_action_dbus_profile_mode� s z Admin._action_dbus_profile_modec C sJ y| j � }| j|� dS tk rD } z| jt|�� dS d }~X nX d S )NTF)rO r] r r r7 )r r\ r! r" r" r# �_action_profile_mode� s zAdmin._action_profile_modec C s> |r,| j j� r:| j j� r:| jd� d}n| jd| � |S )NzCannot enable the tuning.Fz Unable to switch profile: %s)r Z is_running�startr )r �ret�msgr" r" r# �_profile_print_status� s zAdmin._profile_print_statusc C sr t j � | j| j kr.td| j � | jjd�S | jj� rn| j|krn| j sbtd| j � | jjd�S | jjd�S dS )Nz�Operation timed out after waiting %d seconds(s), you may try to increase timeout by using --timeout command line option or using --async.FzError changing profile: %sT) �time� _timestampr r& r rF r ZisSetr r r )r r+ r" r" r# �_action_dbus_wait_profile� s z Admin._action_dbus_wait_profilec C s| | j d ks| j dkrd S y2| jj| j �}d | _ t|dtjd� tjj� W n, tk rv } z| jd� W Y d d }~X nX d S )Nr )�endr% zUError: Failed to stop log capture. Restart the TuneD daemon to prevent a memory leak.) r r Zlog_capture_finishr& r' r( �flushr r )r Zlog_msgsr! r" r" r# r8 s zAdmin._log_capture_finishc C s� t |�dkr| j� S dj|�}|dkr2| jjd�S | jj� | j rn| jd k rn| j d d }| jj | j|�| _ | jj |�\}}| js�| r�| jj| j ||��S tj� | _| jj| j|� | j ||�S )Nr r[ r F� � )�len�_action_dbus_list�joinr rF r �clearr r r �log_capture_startr Zswitch_profilerd re rf r9 rg )r rD r+ r rb rc r" r" r# �_action_dbus_profile s zAdmin._action_dbus_profilec C s< t d� | jjdddg�\}}|dkr0t d� nt d� d S )Nz Trying to (re)start tuned...Zservicer Zrestartr z#TuneD (re)started, changes applied.zQTuneD (re)start failed, you need to (re)start TuneD by hand for changes to apply.)r& r Zexecute)r rb rc r" r" r# �_restart_tuned+ s zAdmin._restart_tunedc C sz || j j� krdy| jj||� | j� dS tk r` } z | jt|�� | jd� dS d }~X qvX n| jd| � dS d S )NTzUnable to switch profile.Fz%Requested profile '%s' doesn't exist.)r Zget_known_namesr Zsave_active_profilerr r r r7 )r r+ rM r! r" r" r# � _set_profile3 s zAdmin._set_profilec C s6 t |�dkr| j� S dj|�}|dkr*dS | j|d�S )Nr r[ r FT)rl rI rn rs )r rD r+ r" r" r# �_action_profileA s zAdmin._action_profilec C s� | j j� }| jj� | j rF| jd k rF| jd d }| j j| j|�| _| j j � \}}| js`| rt| j j | j ||��S t j � | _ | j j| j|� | j ||�S )Nrj rk )r �recommend_profiler ro r r r rp r Z auto_profilerF rd re rf r9 rg )r r+ r rb rc r" r" r# �_action_dbus_auto_profileI s z Admin._action_dbus_auto_profilec C s | j j� }| j|d�S )NF)r � recommendrs )r r+ r" r" r# �_action_auto_profileY s zAdmin._action_auto_profilec C s t | jj� � | jjd�S )NT)r& r ru rF )r r" r" r# � _action_dbus_recommend_profile] s z$Admin._action_dbus_recommend_profilec C s t | jj� � dS )NT)r& r rw )r r" r" r# �_action_recommend_profilea s z Admin._action_recommend_profilec C sr |r| j j� }n | j j� }|r(td� n0td� td� td� td� td� td� tdtj � | j j|�S ) NzIVerification succeeded, current system settings match the preset profile.zLVerification failed, current system settings differ from the preset profile.z=You can mostly fix this by restarting the TuneD daemon, e.g.:z systemctl restart tuned�orz service tuned restartzNSometimes (if some plugins like bootloader are used) a reboot may be required.z&See TuneD log file ('%s') for details.)r Z verify_profile_ignore_missingZverify_profiler& r ZLOG_FILErF )r �ignore_missingrb r" r" r# �_action_dbus_verify_profilee s z!Admin._action_dbus_verify_profilec C s t d� dS )Nz Not supported in no_daemon mode.F)r& )r r| r" r" r# �_action_verify_profilev s z Admin._action_verify_profilec C s: d}| j j| j|�| _| j j� }|s.| jd� | j j|�S )Nrj rk z Cannot disable active profile.r> )r rp r r Zoffr rF )r r rb r" r" r# �_action_dbus_offz s zAdmin._action_dbus_offc C s t d� dS )Nz Not supported in no_daemon mode.F)r& )r r" r" r# � _action_off� s zAdmin._action_offrD c C s( |dkr| j � S |dkr$| j|d�S dS )aD Print accessible profiles or plugins got from TuneD dbus api Keyword arguments: list_choice -- argument from command line deciding what will be listed verbose -- if True then list plugin's config options and their hints if possible. Functional only with plugin listing, with profiles this argument is omitted rD �plugins)�verboseN)rG �_action_dbus_list_plugins)r � list_choicer� r" r" r# rm � s zAdmin._action_dbus_listc C s( |dkr| j � S |dkr$| j|d�S dS )aa Print accessible profiles or plugins with no daemon mode Keyword arguments: list_choice -- argument from command line deciding what will be listed verbose -- Plugins cannot be listed in this mode, so verbose argument is here only because argparse module always supplies verbose option and if verbose was not here it would result in error rD r� )r� N)rI �_action_list_plugins)r r� r� r" r" r# � _action_list� s zAdmin._action_listc C s� | j j� }xv|j� D ]j}t|� | st|| �dkr8q| j j|�}x8|| D ],}td| � |j|d�}|rNtd| � qNW qW | j jd�S )zvPrint accessible plugins Keyword arguments: verbose -- if is set to True then parameters and hints are printed r z %sNz %sT)r Z get_plugins�keysr& rl Zget_plugin_hints�getrF )r r� r� �pluginZhintsZ parameterZhintr" r" r# r� � s z Admin._action_dbus_list_pluginsc C s t d� dS )Nz Not supported in no_daemon mode.F)r& )r r� r" r" r# r� � s zAdmin._action_list_pluginsc C s0 | j j||�\}}|s$| jd| � | j j|�S )Nz Unable to acquire devices: %s)r Zinstance_acquire_devicesr rF )r �devices�instancerb rc r" r" r# �%_action_dbus_instance_acquire_devices� s z+Admin._action_dbus_instance_acquire_devicesc C s t d� dS )Nz Not supported in no_daemon mode.F)r& )r r� r� r" r" r# � _action_instance_acquire_devices� s z&Admin._action_instance_acquire_devicesc C s^ | j j|�\}}}|s0| jd| � | j jd�S x |D ]\}}td||f � q6W | j jd�S )Nz Unable to list instances: %sFz%s (%s)T)r Z get_instancesr rF r& )r � plugin_namerb rc Zpairsr� r� r" r" r# �_action_dbus_get_instances� s z Admin._action_dbus_get_instancesc C s t d� dS )Nz Not supported in no_daemon mode.F)r& )r r� r" r" r# �_action_get_instances� s zAdmin._action_get_instancesc C sR | j j|�\}}}|s0| jd| � | j jd�S x|D ] }t|� q6W | j jd�S )NzUnable to list devices: %sFT)r Zinstance_get_devicesr rF r& )r r� rb rc r� Zdevicer" r" r# �!_action_dbus_instance_get_devices� s z'Admin._action_dbus_instance_get_devicesc C s t d� dS )Nz Not supported in no_daemon mode.F)r& )r r� r" r" r# � _action_instance_get_devices� s z"Admin._action_instance_get_devicesN)r )r )rD F)rD F)F)F)7�__name__� __module__� __qualname__r Z ADMIN_TIMEOUT�loggingZERRORr$ r r r4 r< rA rG rI rK rN rO rQ rR rV rW rX rY rZ rE rH r] r_ r` rd rg r8 rq rr rs rt rv rx ry rz r} r~ r r� rm r� r� r� r� r� r� r� r� r� r" r" r" r# r sd r )Z __future__r Z tuned.adminr Ztuned.utils.commandsr Ztuned.profilesr r � exceptionsr Ztuned.exceptionsr Z tuned.constsr Z tuned.utils.profile_recommenderr r- r' r1 re r r� �objectr r" r" r" r# �<module> s