OwlCyberSecurity - MANAGER
Edit File: manager.cpython-36.pyc
3 �<�e� � @ s~ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlj Z d dl m Z d dl m Z ejj� ZdgZG dd� de�ZdS )� N)� GlobalConfig)�commands�Managerc s� e Zd ZdZd � fdd� Zedd� �Zedd� �Zed d � �Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zejfd d �Z� ZS ) r z? Manager creates plugin instances and keeps a track of them. Nc sJ t t| �j� || _|| _|| _|| _g | _g | _|p:t � | _ t � | _ d S )N) �superr �__init__�_plugins_repository�_monitors_repository�_def_instance_priority�_hardware_inventory� _instances�_pluginsr �_configr �_cmd)�self�plugins_repositoryZmonitors_repositoryZdef_instance_priorityZhardware_inventory�config)� __class__� � /usr/lib/python3.6/manager.pyr s zManager.__init__c C s | j S )N)r )r r r r �plugins! s zManager.pluginsc C s | j S )N)r )r r r r � instances% s zManager.instancesc C s | j S )N)r )r r r r r ) s zManager.plugins_repositoryc C sF |j d krdS | jjtj�}|d kr0| jjd�}tj|j |tj �d k S )NTz /proc/cpuinfo) Z cpuinfo_regexr �get�constsZCFG_CPUINFO_STRINGr Z read_file�re�search� MULTILINE)r �unitZcpuinfo_stringr r r �_unit_matches_cpuinfo- s z Manager._unit_matches_cpuinfoc C sH |j d krdS | jjtj�}|d kr2djtj� �}tj |j |tj �d k S )NT� ) Z uname_regexr r r ZCFG_UNAME_STRING�join�os�unamer r r )r r Z uname_stringr r r �_unit_matches_uname6 s zManager._unit_matches_unamec C s2 g }x�t |j� �D ]�\}}|js0tjd| � q| j|�sJtjd| � q| j|�sdtjd| � q|jjd| j � t |jd �|jd< |j |� qW |j dd� d� t j� }x"|D ]}|jjd� d ||j< q�W x�t |j� �D ]�\}}y$| jj|�}|||< | jj |� W q� tjjjk �r8 tjd| � w�Y q� tk �rt } z tjd | � tj| � w�W Y d d } ~ X q�X q�W g } xf|D ]^}||j }|d k�r��q�tjd |j |jf � |j |j |j |j |j |j!|j�} | j | � �q�W x0| D ](} | j"j#� | j"j$| � | j"j%| � �q�W | j&j'� | j(j)| � d S ) Nz skipping disabled instance '%s'z.skipping instance '%s', cpuinfo does not matchz,skipping instance '%s', uname does not match�priorityc S s | j d S )Nr# )�options)�xr r r �<lambda>P s z Manager.create.<locals>.<lambda>)�keyz2skipping plugin '%s', not supported on your systemz failed to initialize plugin %szcreating '%s' (%s))*�list�itemsZenabled�log�debugr r" r$ � setdefaultr �int�append�sort� collections� OrderedDict�pop�typer �creater �tunedr � exceptionsZNotSupportedPluginException�info� Exception�errorZ exception�nameZcreate_instanceZdevicesZdevices_udev_regexZ script_preZ script_post�pluginZ init_devicesZassign_free_devicesZinitialize_instancer Zstart_processing_eventsr �extend) r Zinstances_configZinstance_info_listZ instance_nameZ instance_infoZplugins_by_nameZ plugin_nameZnoner; �er Z new_instance�instancer r r r4 ? s\ zManager.createc O sX y |||�S t k rR } z,tj� }tjd|t|�f � tj|� |S d }~X nX d S )Nz"BUG: Unhandled exception in %s: %s)r8 � traceback� format_excr* r9 �str)r ZcallerZexc_ret�f�args�kwargsr= Ztracer r r � _try_callu s zManager._try_callc C s� x2| j D ](}tjd|j � | jdd |jj|� qW x.| jD ]$}tjd|j � | jdd |j� q<W | jd d �= | j d d �= d S )Nzdestroying instance %s� destroy_allzcleaning plugin '%s') r r* r+ r: rE r; Zdestroy_instancer Zcleanup)r r> r; r r r rF s zManager.destroy_allc C s4 x.| j jD ]"}tjd| � | jdd |j� q W d S )Nzupdating monitor %s�update_monitors)r Zmonitorsr* r+ rE �update)r Zmonitorr r r rG � s zManager.update_monitorsc C s$ x | j D ]}| jdd |j� qW d S )N� start_tuning)r rE Z apply_tuning)r r> r r r rI � s zManager.start_tuningc C s6 d}x,| j D ]"}| jdd|j|�}|dkr d}q W |S )NT� verify_tuningF)r rE rJ )r Zignore_missing�retr> �resr r r rJ � s zManager.verify_tuningc C s$ x | j D ]}| jdd |j� qW d S )N� update_tuning)r rE rM )r r> r r r rM � s zManager.update_tuningc C s4 | j j� x$t| j�D ]}| jdd |j|� qW d S )N� stop_tuning)r Zstop_processing_events�reversedr rE Zunapply_tuning)r Zrollbackr> r r r rN � s zManager.stop_tuning)N)�__name__� __module__� __qualname__�__doc__r �propertyr r r r r" r4 rE rF rG rI rJ rM r Z ROLLBACK_SOFTrN � __classcell__r r )r r r s 6 )r0 r r r? Ztuned.exceptionsr5 Z tuned.logsZtuned.plugins.exceptionsZ tuned.constsr Ztuned.utils.global_configr Ztuned.utils.commandsr Zlogsr r* �__all__�objectr r r r r �<module> s